diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 3b21a209..de5433f6 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -453,7 +453,7 @@ class MetaFormerBlock(nn.Module): self.drop_path2( self.mlp(self.norm2( x.permute(0, 3, 1, 2) - ).permute(0, 2, 3, 1) + )#.permute(0, 2, 3, 1) ) ) )