RuntimeError: CUDA error: out of memory

RuntimeError: CUDA out of memory. Tried to allocate 78.00 MiB (GPU 0; 2.00 GiB total capacity; 1.07 GiB already allocated; 10.30 MiB free; 1.19 GiB reserved in total by PyTorch)

一般都是batch_size的问题,注意如果修改的是batch_size的default,但是调用时候,batch_size值自己又重新传参了,还是会有问题

你可能感兴趣的:(笔记,pytorch)