0
点赞
收藏
分享

微信扫一扫

使用yolo训练时Loss变为nan[解决思路]


首先参考这篇文章:

​​原创|使用caffe训练时Loss变为nan的原因​​:

  • 梯度爆炸
  • 不当的损失函数
  • 不当的输入
  • 池化层中步长比核的尺寸大

检查自己的train.py:

  1. 检查代码(正确)
  2. 检查输入(自己之前调试的时候修改了图片预处理的一个函数get_random_data)将random=False改成random=True后就可以了
举报

相关推荐

0 条评论