diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index f909b079..2e6a59af 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -735,6 +735,7 @@ class MetaFormer(nn.Module): def forward(self, x): x = self.forward_features(x) + print(x.shape) x = self.forward_head(x) return x