解决RuntimeError: CUDA out of memory. Tried to allocate 236.00 MiB (GPU 0; 11.00

原因是epoch和batch_size太大机器受不了!

epoch全部调整为1,batch_size全部赋值调整为1。

猜你喜欢

转载自blog.csdn.net/com_fang_bean/article/details/107509446