Traceback (most recent call last):
File "train.py", line 621, in <module>
train(hyp, opt, device, tb_writer)
File "train.py", line 89, in train
model = Model(opt.cfg or ckpt['model'].yaml, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device) # create
File "/home/admina/llj/yolov7-main/models/yolo.py", line 546, in __init__
m.stride = torch.tensor([s / x.shape[-2] for x in self.forward(torch.zeros(1, ch, s, s))]) # forward
File "/home/admina/llj/yolov7-main/models/yolo.py", line 601, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "/home/admina/llj/yolov7-main/models/yolo.py", line 627, in forward_once
x = m(x) # run
File "/home/admina/llj/yolov7-main/venv/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/admina/llj/yolov7-main/models/common.py", line 64, in forward
return torch.cat(x, self.d)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 15 but got size 16 for tensor number 1 in the list.