0
点赞
收藏
分享

微信扫一扫

[深度学习][原创]yolov5在GTX1660Ti上训练nan解决方法

雷亚荣 2022-01-31 阅读 98

训练时候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即可解决问题

举报

相关推荐

0 条评论