这会造成什么问题?
一般来说,训练时打印的loss为nan和mAP为0,虽然能保存模型,但训练完也无法进行detect
如何解决?
如果你是CUDA11,降级到CUDA10.2,然后重装Pytorch1.9.0+cu102试试。
并且根据网上资料,Windows环境下在使用detect.py时可能有问题,请尽量在Linux或者MacOS使用yolov5。当然我目前没发现windows下有什么问题
一般来说,训练时打印的loss为nan和mAP为0,虽然能保存模型,但训练完也无法进行detect
如果你是CUDA11,降级到CUDA10.2,然后重装Pytorch1.9.0+cu102试试。
并且根据网上资料,Windows环境下在使用detect.py时可能有问题,请尽量在Linux或者MacOS使用yolov5。当然我目前没发现windows下有什么问题
评论区