YOLOv5s GTX 1660 Ti训练时出现,box,obj,cla全是nan的问题P、R、mAP都是0,Pytorch和cuda、cudnn版本不对
cudnn版本:8.2.2(或者8.2.4)训练显示box、obj、cls都是nan。pytorch版本:1.9.0。Pytorch版本:1.9.0。Pytorch版本:1.9.0。cudnn版本:8.0.5。cudnn版本:8.2.2。cuda版本:11.1。cuda版本:11.4。cuda版本:10.2。
原创
2023-05-04 03:56:29 ·
1272 阅读 ·
1 评论