YOLOv7-Pytorch-Segmentation icon indicating copy to clipboard operation
YOLOv7-Pytorch-Segmentation copied to clipboard

I had problems training the coco128 dataset

Open 1063274739 opened this issue 2 years ago • 0 comments

Traceback (most recent call last): File "train.py", line 630, in main(opt) File "train.py", line 526, in main train(opt.hyp, opt, device, callbacks) File "train.py", line 308, in train loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size File "./utils/loss.py", line 125, in call tcls, tbox, indices, anchors = self.build_targets(p, targets) # targets File "./utils/loss.py", line 198, in build_targets anchors, shape = self.anchors[i], p[i].shape AttributeError: 'list' object has no attribute 'shape' I hope you can have a look. Thank you very much

1063274739 avatar Nov 10 '22 14:11 1063274739