From dd57cde1bc3905a4c724184d6bd5509bba4834c4 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Fri, 20 Jan 2023 01:56:03 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) ) ) )