【计算机视觉】YOLOv5训练报错：RuntimeError: result type Float can‘t be cast to the desired output type __int64

一、问题提出

在使用yolov5训练自己的数据集时，运行train.py时出现上述问题：

Image sizes 1536 train, 1536 val
Using 2 dataloader workers
Logging results to runs\train\exp6
Starting training for 100 epochs...

     Epoch   gpu_mem       box       obj       cls    labels  img_size
  0%|                                                                                                                                                                    | 0/3236 [00:00<?, ?it/s] 
Traceback (most recent call last):
  File "train.py", line 630, in <module>
    main(opt)
  File "train.py", line 527, in main
    train