einops.EinopsError

[Michael-Chao]

import torch from light_cnns import mobilevit_s model = mobilevit_s() model.eval() print(model) input = torch.randn(1, 3, 224, 224) y = model(input) print(y.size()) 这段代码输入报错：einops.EinopsError: Error while processing rearrange-reduction pattern "b d (h ph) (w pw) -> b (ph pw) (h w) d". Input tensor shape: torch.Size([1, 240, 7, 7]). Additional info: {'ph': 2, 'pw': 2}. Shape mismatch, can't divide axis of length 7 in chunks of 2

修改为（1, 3, 256, 256）才好