search

YunYang1994 / tensorflow-yolov3

🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"
https://yunyang1994.gitee.io/2018/12/28/YOLOv3-算法的一点理解/
MIT License
3.64k stars 1.36k forks source link

6G显存只能batchsize=2 #485

Open JokerKangS opened 4 years ago

JokerKangS commented 4 years ago

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

yanxi00 commented 4 years ago

我1660s只能1

wyY0105 commented 3 years ago

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