最近yolov10更新了,我也是打算去体验一下这个新玩意。按网上的教程把环境配置好后,跑起来却出现了训练时检测出现nan值的情况(如下图),很生气
这个问题我在用yolov5时其实就已经遇到了,当时在网上到处查教程,最后的解决方案是用cuda10.2的版本,虽然速度慢了一些,但是好歹能跑起来了。这次yolov10又出这个岔子,可是yolov10要求的torch版本对应的cuda版本最低都是11.7版本了,降级这一招行不通了(醉了)。我就又去网上查相关解决方法,不过都没有yolov10这个版本的对应的解决方法。没办法,我只能参考yolov8版本对应的解决方案,琢磨出了一个yolov10版本的解决方法,这个方法就是:打开yolov10项目路径下的ultralytics/cfg/default.yaml,打开该文件,搜索amp,将其参数改为False,如下图所示:
然后就可以愉快的开始训练我的人工智障啦: