alterzero / DBPN-Pytorch

The project is an official implement of our CVPR2018 paper "Deep Back-Projection Networks for Super-Resolution" (Winner of NTIRE2018 and PIRM2018)
https://alterzero.github.io/projects/DBPN.html
MIT License
566 stars 164 forks source link

DBPN_x8.pth load error #21

Closed suke27 closed 5 years ago

suke27 commented 6 years ago

Hello, when I try to use DBPN_x8 model below error occurs RuntimeError: invalid argument 2: sizes do not match at /pytorch/torch/lib/THC/generic/THCTensorCopy.c:48

DBPN_x8.pth is it not for x8, any difference with NTIRE2018_x8.pth?

Liber-coder commented 6 years ago

I encountered similar problem when I changed the --model from NTIRE2018_x8 to DBPN_x8 in the file eval.py. And I solved the problem after changing --model_type from DBPNNL to DBPN or other.

I hope this method is also useful for you. :)

suke27 commented 6 years ago

yes, I also found it. thank you!