目标检测小白 2022-08-12 13:39 采纳率: 25%
浏览 504
已结题

yolov5训练损失为Nan

问题遇到的现象和发生背景

yolov5中Bottleneck中添加注意力机制,当使用yolov5s结构时能正常运行,当使用网络深度更深的yolov5m时训练变为Nan

  • 写回答

1条回答 默认 最新

  • 脚踏南山 2022-08-12 13:55
    关注

    将学习率改为0.00001,步子太大,扯到蛋了,改小点

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 8月20日
  • 已采纳回答 8月12日
  • 创建了问题 8月12日