unexpected key(s) in state_dict: "params"

xinntao / ESRGAN

ECCV18 Workshops - Enhanced SRGAN. Champion PIRM Challenge on Perceptual Super-Resolution. The training codes are in BasicSR.

https://github.com/xinntao/BasicSR

Apache License 2.0

5.86k stars 1.04k forks source link

unexpected key(s) in state_dict: "params" #110

Open Esprovas opened 3 years ago

Esprovas commented 3 years ago

Hi Xinntao, I tried to run the test.py but I get this error.. tried old models also the new and the same thing.. I am using Pytorch, Anaconda3 and has last CUDA installed all latest versions..

Error picture: https://puu.sh/Gu8HF/a83bd8daf9.png

Esprovas commented 3 years ago

https://puu.sh/Gu8KE/a4d271a637.png

xinntao commented 3 years ago

Hi @Esprovas Sorry for the late reply,

If you use this ESRGAN repo, you need to download from https://drive.google.com/drive/u/0/folders/17VYV_SoZZesU6mbxz2dMAIccSSlqLecY

I think the RRDB_ESRGAN_x4.pth model should be OK.

(For pretrained models with 'params' are the latest models for BasicSR repo).

Sorry for the incompatible issues.