yolov8在设置amp=False 之后map 训练依旧为0 解决办法

可能原因 是cuda 版本导致的半精度浮点数计算出现nan的bug

  1. 解决办法 设置amp=False 就是不使用混合精度训练。
  2. 或者直接改用低版本的cuda和pytorch。cuda11.6 以下 直接有效
  3. 也有可能是学习率过高 降低学习率

设置amp=False之后还是存在问题 是因为yolov8库的问题 按以下修改
找到torch_utils.py 修改425行 去掉 half()
yolov8在设置amp=False 之后map 训练依旧为0 解决办法_第1张图片

你可能感兴趣的:(神经网络,YOLO)