基于Windows系统进行Yolov3训练模型遇到：“CUDA Error: out of memory”的解决方案

基于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

 

 

成功解决：

 

如若还不行，基于这三个参数略微变动，即有大概率可解决。