GPU out of memory !
Opened this issue · 1 comments
nabil215 commented
RuntimeError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 4.00 GiB total capacity; 2.77 GiB already allocated; 115.52 MiB free; 2.78 GiB reserved in total by PyTorch)
GuillaumeMacaigne commented
By default, images seems to be loaded with their default size. Try setting arg.img_size and arg.val_img_size to a lower value