Yolov5 模型训练过程中报错：RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

Question

• 操作系统 ：win
  python版本：python3.7.11
  pytorch版本 ：1.8.1
  cuda：11
  ##cmd 在yolo目录下 输入

'python train.py --img 640 --batch 8 --epochs 300 --data ../datasets/data.yaml --weights weights/yolov5s.pt --cache'

##开始训练模型

Traceback (most recent call last):
  File "train.py", line 463, in <module>
    train(hyp, opt, device, tb_writer)
  File "train.py", line 286, in train
    loss, loss_items = compute_loss(pred, targets.to(device), model)  # scale
  File "C:\Users\13148238788\Desktop\yolov5\utils\utils.py", line 501, in com
    return loss * bs, torch.cat((lbox, lobj, lcls, loss)).detach()
RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

报以上错误
目前暂无思路

Answer 1

1.首先检查一下数据集，看下数据集路径是否正确，图片和标签的数据是不是正确的
2.上面没问题的话就去下载最新版本看下
3.yolov5的话要求好像是py3.8，你的python3.7可能会有影响。