yolov5版本:6.1
环境:torch 2.0.0+cu118
显卡:GTX1650
开始时训练集大小为263,batch_size大小为4
训练出现以下问题:box、obj、cls全为nan,P、R、map全为0
试着将训练集大小改为260后问题解决。
如果觉得删除环境再重装比较麻烦的话,可以先试着修改训练集大小或者batch_size大小,令训练集大小为batch_size的整数倍,看看是否能解决问题。
如果不能解决问题,可以参考其他回答:解决YOLOV5出现全为nan和0的问题_yolov5训练时 gpu_men为0-CSDN博客yolo系列算法训练时loss出现nan值,解决办法(GTX16xx系列显卡的问题)_yolo训练需要什么显卡-CSDN博客