你好，最新版的yoloair在训练acmix的时候报如下错误

[darkredrock]

Traceback (most recent call last):
File "E:\code\yolov5\yoloair-main\train.py", line 690, in
main(opt)
File "E:\code\yolov5\yoloair-main\train.py", line 586, in main
train(opt.hyp, opt, device, callbacks)
File "E:\code\yolov5\yoloair-main\train.py", line 125, in train
model = Model(cfg or ckpt['model'].yaml, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device) # create
File "E:\code\yolov5\yoloair-main\models\yolo.py", line 132, in init
m.stride = torch.tensor([s / x.shape[-2] for x in self.forward(torch.zeros(1, ch, s, s))]) # forward
File "E:\code\yolov5\yoloair-main\models\yolo.py", line 165, in forward
return self._forward_once(x, profile, visualize) # single-scale inference, train
File "E:\code\yolov5\yoloair-main\models\yolo.py", line 188, in _forward_once
x = m(x) # run
File "D:\Program Files\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "E:\code\yolov5\yoloair-main\models\common.py", line 850, in forward
unfold_k = self.unfold(self.pad_att(k_att)).view(b * self.head, self.head_dim,
File "D:\Program Files\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "D:\Program Files\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\padding.py", line 174, in forward
return F.pad(input, self.padding, 'reflect')
File "D:\Program Files\anaconda3\envs\pytorch\lib\site-packages\torch\nn\functional.py", line 4189, in _pad
return torch._C._nn.reflection_pad2d(input, pad)
RuntimeError: Argument #4: Padding size should be less than the corresponding input dimension, but got: padding (256, 256) at dimension 3 of input 4

[iscyy]

You should update the latest code

[Leondre1]

Please，how to solve this problem？ I have the same problem.