search

jixiaozhong / RealSR

Real-World Super-Resolution via Kernel Estimation and Noise Injection
Apache License 2.0
740 stars 102 forks source link

Change batch size in test.py #35

Open shiva811 opened 3 years ago

shiva811 commented 3 years ago

Hello All I am trying to run test.py on some images captured by my DSLR, CUDA_VISIBLE_DEVICES=1 python3 test.py -opt options/df2k/test_df2k.yml

and I am always getting CUDA out of memory error. However, when I try to run it on the FLICKR dataset of DIV2K, I am not facing any such problems. Please can anybody help me resolve it?

I think if i reduce the batch size, then it must be able to resolve this issue, can anybody tell me where the batch-size option is? If you think it could be because of something else also, please let me know.

This is the error I am getting always

RuntimeError: CUDA out of memory. Tried to allocate 10.23 GiB (GPU 0; 23.65 GiB total capacity; 13.52 GiB already allocated; 5.71 GiB free; 14.81 GiB reserved in total by PyTorch)

Thank you in advance for your help.