可能原因 是cuda 版本导致的半精度浮点数计算出现nan的bug
- 解决办法 设置amp=False 就是不使用混合精度训练。
- 或者直接改用低版本的cuda和pytorch。cuda11.6 以下
直接有效
- 也有可能是学习率过高 降低学习率
设置amp=False之后还是存在问题 是因为yolov8库的问题 按以下修改
找到torch_utils.py 修改425行 去掉 half()
可能原因 是cuda 版本导致的半精度浮点数计算出现nan的bug
直接有效
设置amp=False之后还是存在问题 是因为yolov8库的问题 按以下修改
找到torch_utils.py 修改425行 去掉 half()
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1028979.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!