运行环境

atlas         3000 

torch          2.1.0

torch-npu     2.1.0

torchvision     0.16.0

cann         Ascend-cann-toolkit_7.0.0

yolo   modelzoo/Yolov5_for_PyTorch_v7.0

报错信息如下