Yolov3 出现avg loss nan的情况

Note: If during training you see nan values for avg (loss) field - then training goes wrong, but if nan is in some other lines - then training goes well.
作者原话,但我依然在avg loss为nan时训练了8100次,结果探测时没有任何结果。

反复检查voc_label.py,yolov3.cfg,myv3.data,myv3.names文件,改变打标策略,改变检测种类数量到1,avg loss还是为nan。

结果最后使用之前训练我所需要检测图片已经训练过的权重当做detector train的起始权重,不用yolov3 darknet作者提供的darknet53.conv.74权重文件,avg loss正常下降。说明起始权重所对应图片集与我所要训练的图片集若有较大差异,训练会出现问题。

猜你喜欢

转载自blog.csdn.net/weixin_42388228/article/details/83178513
今日推荐