之前在用yolo v3训练自己的数据集的时候,会出现loss=nan的情况。这边给出一点解决方法。
1.查看是否为代码问题,在计算损失时是否出现负数,分母为0等情况。
2.检查数据集文件是否标识正确。
3.每一次batch,打印一次loss,检查是否出现梯度爆炸的情况。若有loss=inf,则可以设定一个阈值。
4.调整参数,我就是因为学习率过小出现了nan值,调大一点就正常了
之前在用yolo v3训练自己的数据集的时候,会出现loss=nan的情况。这边给出一点解决方法。
1.查看是否为代码问题,在计算损失时是否出现负数,分母为0等情况。
2.检查数据集文件是否标识正确。
3.每一次batch,打印一次loss,检查是否出现梯度爆炸的情况。若有loss=inf,则可以设定一个阈值。
4.调整参数,我就是因为学习率过小出现了nan值,调大一点就正常了
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。