acmix."RuntimeError: Padding size should be less than the corresponding input dimension, but got: padding (3, 3) at dimension 2 of input [16, 128, 3, 6]"

[martinkeith]

Search before asking

• [X] I have searched the YOLOAir issues and found no similar questions.

Question

when I used the "yolo5s_acmix.yaml" from the project. I added the commond option " parser.add_argument('--acmix', action='store_true', help='useacmix')" in train.py and added the val.run(... half = not opt.acmix) in train.py

use the commond: python train.py --cfg ./configs/attention/yolovs_acmix.yaml --acmix

I got the error:

RuntimeError: Padding size should be less than the corresponding input dimension, but got: padding (3, 3) at dimension 2 of input [16, 128, 3, 6]

so, what's wrong with me?

Additional

No response

[martinkeith]

I got it. just change the image size from 160 to 640(or other size)