基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案
遇到问题:CUDA Error: out of memory
原因:爆显存 Hhhhhha !!!
解决方案:更改训练所使用的CFG(.cfg)文件参数 (本人使用的是:yolov3-voc.cfg)
(1)修改 batch subdivisions 值 (batch可较默认值64略小,subdivisions可
较默认值48略大),此处设置:batch=48, subdivisions=48
(2)修改尾行 random 值 (默认值为1),此处修改为 :random=0
成功解决:
如若还不行,基于这三个参数略微变动,即有大概率可解决。