Revert "Update metaformers.py"

This reverts commit 2916f37f8d.
2 years ago · e4f89b2e25
parent 2916f37f8d
commit e4f89b2e25
1 changed files with 2 additions and 4 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -816,10 +816,8 @@ 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