search

Holmes-Alan / ABPN

Attention based Back Projection Network (ABPN) for image ultra-resolution in ICCV2019
75 stars 20 forks source link

Memory error #4

Open BeepBump opened 5 years ago

BeepBump commented 5 years ago

Thank you for the great work and sharing the code! However, when I adopt space attention in other SR models, there is no problem in training, but will have memory error in validation phase. Can u help me solve this problem? Thanks!

Holmes-Alan commented 5 years ago

you have check your image size and patch size you use for testing. The maximum patch size is 64

BeepBump commented 5 years ago

Then, if my testing images are larger than 64x64, how could I do to test on these datasets?

Holmes-Alan commented 5 years ago

just run the testing code. It super-resolve the image patch by patch for reconstruction.