Файл "/program/multiclassification/classify/yolov8/yolo_dalei.py", строка 212, в функции detect_image:
outputs = self.net(images)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1511, в функции _wrapped_call_impl:
return self._call_impl(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1520, в функции _call_impl:
return forward_call(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", строка 183, в функции forward:
return self.module(*inputs[0], **module_kwargs[0])
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1511, в функции _wrapped_call_impl:
return self._call_impl(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1520, в функции _call_impl:
return forward_call(*args, **kwargs)
Файл "/program/multiclassification/classify/yolov8/nets/yolo.py", строка 124, в функции forward:
feat1, feat2, feat3 = self.backbone.forward(x)
Файл "/program/multiclassification/classify/yolov8/nets/backbone.py", строка 126, в функции forward:
x = self.stem(x)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1511, в функции _wrapped_call_impl:
return self._call_impl(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1520, в функции _call_impl:
return forward_call(*args, **kwargs)
Файл "/program/multiclassification/classify/yolov8/nets/backbone.py", строка 35, в функции forward_fuse:
return self.act(self.conv(x))
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1511, в функции _wrapped_call_impl:
return self._call_impl(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/module.py", строка 1520, в функции _call_impl:
return forward_call(*args, **kwargs)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/conv.py", строка 460, в функции forward:
return self._conv_forward(input, self.weight, self.bias)
Файл "/usr/local/python38/lib/python3.8/site-packages/torch/nn/modules/conv.py", строка 456, в функции _conv_forward:
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: _Map_base::at