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

  • 操作系统 :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

报以上错误
目前暂无思路

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