tensorflow-yolov3 icon indicating copy to clipboard operation
tensorflow-yolov3 copied to clipboard

Published 20 hours ago • verified publisher icon YunYang1994

6G显存只能batchsize=2

Open JokerKangS opened this issue 5 years ago • 2 comments

我用1660s跑训练，batchsize最大是2，再增加就会显存溢出，有什么办法跑大batch吗，难道6G显存只能跑batchsize=2么，感谢

JokerKangS avatar Feb 29 '20 14:02 JokerKangS

我1660s只能1

yanxi00 avatar May 11 '20 07:05 yanxi00

大佬们，我是1660的显存6g，训练跑到第20个epoch时，就显存报错out of memory停止了,batch_size改成3，还是不行，咋办

wyY0105 avatar Dec 31 '20 03:12 wyY0105