An error is reported during the fpn_efficientformerv2_s0 test.

[nizhenliang]

q = self.q(x).flatten(2).reshape(B, self.num_heads, -1, H * W).permute(0, 1, 3, 2)
RuntimeError: shape '[1, 8, -1, 336]' is invalid for input of size 90112

[nypyp]

I have met the same err, have you solved this?

[nypyp]

I have use the mmsegmentation to run this as backbone, and I solve this err by copy the config file of poolformer, the key is adding this line

dict(type='ResizeToMultiple', size_divisor=32) 

in test_pipeline