bluedream1121/REKD

RuntimeError: CUDA out of memory.

Closed this issue · 0 comments

Hello! I'm very interested in your paper and code. When I try to reproduce the code on Test on the HPatches, I encounter a RuntimeError: CUDA out of memory. Tried to allocate 1.70 GiB (GPU 0; 11.91 GiB total capacity; 8.88 GiB already allocated; 1.32 GiB free; 9.82 GiB reserved in total by PyTorch). How can I resolve this issue?
thanks for your reply!