16系显卡在YOLOv10训练时检测出现nan值的解决方法

最近yolov10更新了,我也是打算去体验一下这个新玩意。按网上的教程把环境配置好后,跑起来却出现了训练时检测出现nan值的情况(如下图),很生气

这个问题我在用yolov5时其实就已经遇到了,当时在网上到处查教程,最后的解决方案是用cuda10.2的版本,虽然速度慢了一些,但是好歹能跑起来了。这次yolov10又出这个岔子,可是yolov10要求的torch版本对应的cuda版本最低都是11.7版本了,降级这一招行不通了(醉了)。我就又去网上查相关解决方法,不过都没有yolov10这个版本的对应的解决方法。没办法,我只能参考yolov8版本对应的解决方案,琢磨出了一个yolov10版本的解决方法,这个方法就是:打开yolov10项目路径下的ultralytics/cfg/default.yaml,打开该文件,搜索amp,将其参数改为False,如下图所示:

然后就可以愉快的开始训练我的人工智障啦:

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值