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

中间件小哥

 关注

阅读 116

2023-01-25


首先参考这篇文章：

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

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

检查自己的train.py：

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

相关推荐

解决：模型训练时loss出现nan

萧萧雨潇潇 194 0 0

训练网络loss出现Nan解决办法

邯唐情感 252 0 0

NAN Loss

司马吹风 246 0 0

loss超大，训练一次后loss为nan

拾光的Shelly 161 0 0

大模型训练中Loss出现NaN的解决策略

Sophia的玲珑阁 57 0 0

keras使用yolov3训练自己的数据时出现- val_loss: nan

_阿瑶 240 0 0

模型训练中出现loss为NaN怎么办？

kolibreath 38 0 0

pytorch中loss出现nan

有态度的萌狮子 65 0 0

【DL基础】训练DL网络loss出现Nan的原因分析

_铁马冰河_ 169 0 0

关于 train loss、val loss训练时遇到的问题

书坊尚 811 0 0

精彩评论（0）

0 0 举报