Revert "Update metaformers.py"

This reverts commit 2fef9006d7.
2 years ago · 47b718a1c8
parent 869e01cec6
commit 47b718a1c8
1 changed files with 1 additions and 1 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -822,7 +822,7 @@ class MetaFormer(nn.Module):
        #x = self.stages(x)
        for i, stage in enumerate(self.stages):
            x=stage(x)
-            print(x[0][0][0][0])
+            print(x)
        
        
        return x