标签搜索

目 录CONTENT

文章目录

解决YOLOV5训练时loss为nan和mAP为0的情况

陈铭
2021-09-07 / 0 评论 / 1 点赞 / 896 阅读 / 126 字 / 正在检测是否收录...

这会造成什么问题?

一般来说,训练时打印的loss为nan和mAP为0,虽然能保存模型,但训练完也无法进行detect

如何解决?

如果你是CUDA11,降级到CUDA10.2,然后重装Pytorch1.9.0+cu102试试。
并且根据网上资料,Windows环境下在使用detect.py时可能有问题,请尽量在Linux或者MacOS使用yolov5。当然我目前没发现windows下有什么问题

1

评论区