雷亚荣

关注

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

雷亚荣

关注

阅读 99

2022-01-31

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

相关推荐

ixiaoyang8

yolov5训练学习笔记1

ixiaoyang8 26 0 0

书写经典

【colab】在colab上使用yolov5训练自己的模型

书写经典 183 0 0

凛冬已至夏日未远

图像检测【YOLOv5】——深度学习

凛冬已至夏日未远 24 0 0

Mhhao

YOLOV5:在本地电脑训练模型

Mhhao 161 0 0

北邮郭大宝

深度学习-Yolov5入门到实战点选数据集训练

北邮郭大宝 121 0 0

RJ_Hwang

《深度学习与目标检测 YOLOv5》

RJ_Hwang 126 0 0

凯约

[深度学习][原创]yolo训练时候提示页面文件太小,无法完成操作5种解决方法

凯约 12 0 0

程序员漫画编程

VisDrone2019上训练YOLOv5(用ultralytics)

程序员漫画编程 55 0 0

外贸达人小峻先森

YoloV5在tensorRT上加速(Ubuntu)(C++)

外贸达人小峻先森 169 0 0

求索大伟

[课程][原创]使用yolov5训练自己实例分割模型windows版

求索大伟 12 0 0

精彩评论(0)

0 0 举报