Traceback (most recent call last):
File "train.py", line 80, in <module>
model = YOLO(r'ultralytics/cfg/models/11/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU.yaml')
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/models/yolo/model.py", line 23, in __init__
super().__init__(model=model, task=task, verbose=verbose)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/engine/model.py", line 143, in __init__
self._new(model, task=task, verbose=verbose)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/engine/model.py", line 251, in _new
self.model = (model or self._smart_load("model"))(cfg_dict, verbose=verbose and RANK == -1) # build model
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/tasks.py", line 378, in __init__
m.stride = torch.tensor([s / x.shape[-2] for x in forward(torch.zeros(1, ch, s, s))]) # forward
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/tasks.py", line 370, in <lambda>
forward = lambda x: self.forward(x)[0] if isinstance(m, (Segment, Pose, OBB)) else self.forward(x)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/tasks.py", line 112, in forward
return self.predict(x, *args, **kwargs)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/tasks.py", line 130, in predict
return self._predict_once(x, profile, visualize, embed)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/tasks.py", line 161, in _predict_once
x = m(x) # run
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/modules/DynamicDCNv3Head.py", line 1376, in forward
x = self.dyhead_tower(tensor_dict)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/container.py", line 204, in forward
input = module(input)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/modules/DynamicDCNv3Head.py", line 1270, in forward
temp_fea.append(F.interpolate(self.DyConv[0](input, **conv_args),
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/modules/DynamicDCNv3Head.py", line 1218, in forward
x = self.dcnv3(input)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/yolov11n-StartNet_C2PSA-SCSA_DyHead-DCNv3_FocalerIoU/ultralytics/nn/modules/DynamicDCNv3Head.py", line 1054, in forward
x = self.input_proj(input)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 114, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (1600x128 and 64x64)
最新发布