Zheng222 / IMDN

Lightweight Image Super-Resolution with Information Multi-distillation Network (ACM MM 2019, Winner Award of ICCVW AIM 2019 Constrained SR Track1&Track2)
425 stars 70 forks source link

EDSR's forward_chop replace the crop_forward #13

Closed zrongcheng closed 4 years ago

zrongcheng commented 4 years ago

Thanks for your excellent works. I have a question. I tested my images(3040x3040 or 6080x3040) with EDSR, little change in gpu memory . but, when i tested your IMDN with forward_chop, why the memory change so much?

Zheng222 commented 4 years ago

@zrongcheng Hello, the basic block of EDSR is different from IMDN, which has nothing to do with crop strategy. You can calculate the Line Buffer of basic blocks of EDSR and IMDN.