chenyuntc/pytorch-best-practice

RuntimeError:cuda runtime error(2):out of memory

lizxchen opened this issue · 1 comments

lizxchen commented

有两个8G的显卡，显示这个错误，想问一下原因，谢谢

foolishflyfox commented

把 config.py 中的 batch_size 改小