Update metaformers.py

2 years ago · 6b6510f30a
parent 808f4a7ebd
commit 6b6510f30a
1 changed files with 4 additions and 1 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -819,7 +819,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)
        
        
        return x