[深度学习][原创]yolov5在GTX1660Ti上训练nan解决方法
Posted FL1623863129
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[深度学习][原创]yolov5在GTX1660Ti上训练nan解决方法相关的知识，希望对你有一定的参考价值。
训练时候box ,obj，cls一直是nan安装任何版本pytorch无效，更换yolov5-6.0最新版本无效
Epoch gpu_mem box obj cls labels img_size
0/273 5.64G nan nan nan 113 640
解决方法：
安装torch==19.1+cu102和torchvision==0.10.1+cu102即可解决问题
以上是关于[深度学习][原创]yolov5在GTX1660Ti上训练nan解决方法的主要内容，如果未能解决你的问题，请参考以下文章
[深度学习][原创]使用labelImg+yolov5完成所有slowfast时空动作检测项目-配置文件篇
[深度学习][原创]旋转目标检测框架yolov5_obb，paddledetection-s2anet和mmrotate谁最好用?
[深度学习][原创]使用labelImg+yolov5完成所有slowfast时空动作检测项目-开山篇
[深度学习][原创]让yolov6-0.1.0支持yolov5的txt读取数据集模式