search

Paper99 / SRFBN_CVPR19

Pytorch code for our paper "Feedback Network for Image Super-Resolution" (CVPR2019)
MIT License
551 stars 126 forks source link

RuntimeError: CUDA error: out of memory #1

Closed zhLawliet closed 5 years ago

zhLawliet commented 5 years ago

when run python test.py -opt options/test/test_SRFBN_example.json the error is below: VPR19-master/networks/srfbn_arch.py", line 79, in forward LD_H = torch.cat(tuple(hr_features), 1) RuntimeError: CUDA error: out of memory

i have run the ESRGAN and SRGAN without the error.

Paper99 commented 5 years ago

https://github.com/Paper99/SRFBN_CVPR19/blob/0b2f7f4d418f6580fd009a397762021d2deeaf1e/solvers/SRSolver.py#L192

You can decrease min_size in this line. I hope it is helpful for you.

zhLawliet commented 5 years ago

thanks, it is ok when i decrease the min_size.