diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index b7435128..d5bf9692 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -809,11 +809,12 @@ class MetaFormer(nn.Module): if pre_logits: return x - x = self.global_pool(x) - x = x.squeeze() - x = self.norm(x) + #x = self.global_pool(x) + #x = x.squeeze() + #x = self.norm(x) # (B, H, W, C) -> (B, C) - x = self.head(x) + #x = self.head(x) + x=self.head(self.norm(x.mean([1, 2]))) return x def forward_features(self, x):