训练时候box ,obj,cls一直是nan安装任何版本pytorch无效,更换yolov5-6.0最新版本无效
Epoch gpu_mem box obj cls labels img_size
0/273 5.64G nan nan nan 113 640
解决方法:
安装torch==1.9.1+cu102和torchvision==0.10.1+cu102即可解决问题