Update metaformers.py

2 years ago · 2916f37f8d
parent 1d882eb494
commit 2916f37f8d
1 changed files with 4 additions and 2 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -816,8 +816,10 @@ class MetaFormer(nn.Module):
    def forward_features(self, x):
        x = self.patch_embed(x)
-        x = self.stages(x)
+        #x = self.stages(x)
-        
+        for i, stage in enumerate(self.stages):
            x = stage(x)
            print(x[0][0][0][0])
        return x