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博客
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
