From e4f89b2e25064fb4368a83cc5bc2d916f6f3a2cf Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Tue, 17 Jan 2023 12:33:31 -0800
Subject: [PATCH] Revert "Update metaformers.py"

This reverts commit 2916f37f8d862cd4fcbc86c858b23894daec3339.
---
 timm/models/metaformers.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index 6f09dd74..a108afda 100644
--- 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)
-        for i, stage in enumerate(self.stages):
-            x = stage(x)
-            print(x[0][0][0][0])
+        x = self.stages(x)
+        
         
         return x