Update metaformers.py

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