Giters

xinntao / Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

keep getting mismatch conv_first weight warning help

xalteropsx opened this issue · comments

commented 
    model2.load_weights(f"weights/RealESRNet_x4plus.pth", download=True)
  File "Z:\software\python11\Lib\site-packages\py_real_esrgan\model.py", line 53, in load_weights
    self.model.load_state_dict(loadnet, strict=True)
  File "Z:\software\python11\Lib\site-packages\torch\nn\modules\module.py", line 2191, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for RRDBNet:
        size mismatch for conv_first.weight: copying a param with shape torch.Size([64, 12, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 3, 3, 3]).