程序问题

[01SHENyankun]

RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 4.00 GiB total capacity; 2.48 GiB already allocated; 0 bytes free; 2.77 GiB reserved in total by PyTorch)
报错的是这样的，我尝试把batch_size调小也没有解决问题，在网上说加with torch.no_grad():,但是我不知道加在哪里，所以请教一下，这是我复现的第一篇论文，意义重大

[daa233]

你用的什么显卡呀？看上去就是显存不够了

[01SHENyankun]

NVIDIA GeForce GTX 1650

[01SHENyankun]

这种问题有其他解决办法吗

[daa233]

看上去总显存是4G，可以试试 batch_size 1 能不能跑起来。如果能跑起来，就可以用小 batch size 跑。

[01SHENyankun]

不知道是不是我代码写的不对，我是这么写的batch_size = int(real_image.size(0)/16),我试了试2，4，8，32，结果又报错了，说要解压的值太多