Change batch size in test.py

[shiva811]

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.