File "E:\Python_Project\yolov5-6.2\yolov5-6.2\models\yolo.py", line 105, in _forward_once
x = m(x) # run
File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, kwargs)
File "E:\Python_Project\yolov5-6.2\yolov5-6.2\models\common.py", line 1115, in forward
unfold_k = self.unfold(self.pad_att(k_att)).view(b self.head, self.head_dim,
File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(input, kwargs)
File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\padding.py", line 174, in forward
return F.pad(input, self.padding, 'reflect')
RuntimeError: Argument #4: Padding size should be less than the corresponding input dimension, but got: padding (512, 512) at dimension 3 of input 4
File "E:\Python_Project\yolov5-6.2\yolov5-6.2\models\yolo.py", line 105, in _forward_once x = m(x) # run File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl return forward_call(*input, kwargs) File "E:\Python_Project\yolov5-6.2\yolov5-6.2\models\common.py", line 1115, in forward unfold_k = self.unfold(self.pad_att(k_att)).view(b self.head, self.head_dim, File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl return forward_call(input, kwargs) File "E:\anaconda\envs\xiaozhai\lib\site-packages\torch\nn\modules\padding.py", line 174, in forward return F.pad(input, self.padding, 'reflect') RuntimeError: Argument #4: Padding size should be less than the corresponding input dimension, but got: padding (512, 512) at dimension 3 of input 4