generative-inpainting-pytorch icon indicating copy to clipboard operation
generative-inpainting-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon daa233

程序问题

Open 01SHENyankun opened this issue 3 years ago • 5 comments

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():,但是我不知道加在哪里，所以请教一下，这是我复现的第一篇论文，意义重大

01SHENyankun avatar Sep 16 '21 03:09 01SHENyankun

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

daa233 avatar Sep 16 '21 05:09 daa233

NVIDIA GeForce GTX 1650

01SHENyankun avatar Sep 16 '21 06:09 01SHENyankun

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

01SHENyankun avatar Sep 16 '21 06:09 01SHENyankun

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

daa233 avatar Sep 16 '21 07:09 daa233

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

01SHENyankun avatar Sep 16 '21 07:09 01SHENyankun