J.yang comments

Repositories
Issues
Comments

Results 2 comments of


                                            J.yang

MobileViT-einops.EinopsError: Error while processing rearrange-reduction pattern "b d (h ph) (w pw) -> b (ph pw) (h w) d".

In External-Attention-pytorch/model/backbone/MobileViT.py，line134，you can change it. `nn.Conv2d(inp, hidden_dim, kernel_size=1, stride=self.stride, bias=False),`

MobileViT-einops.EinopsError: Error while processing rearrange-reduction pattern "b d (h ph) (w pw) -> b (ph pw) (h w) d".

看起来像是你的输入数据的问题，你可以先定义一个随机数，然后调试一下，观察每一层的输出结果是否正确。 `input = torch.randn(1, 3, 256, 256)`