'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可能会有影响。