Update metaformers.py

2 years ago · cc8575e9d1
parent bde9554604
commit cc8575e9d1
1 changed files with 1 additions and 0 deletions
--- 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