issue about train the model on DIV2K

Paper99 / SRFBN_CVPR19

Pytorch code for our paper "Feedback Network for Image Super-Resolution" (CVPR2019)

MIT License

551 stars 126 forks source link

issue about train the model on DIV2K #35

Open zhiminghao95 opened 5 years ago

zhiminghao95 commented 5 years ago

I find the bug

Sizes of tensors must match except in dimension 0. Got 39 and 40 in dimension 2 at c:\n ew-builder_3\win-wheel\pytorch\aten\src\th\generic/THTensorMath.cpp:3616

what should I do to solve the bug? thank you

Paper99 commented 5 years ago

You can first check the dataset options in your .json file.