Update metaformers.py

2 years ago · cf28c57fda
parent 2fef9006d7
commit cf28c57fda
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])
+            print(x[0][0][0][0])
        return x