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

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

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index a108afda..6f09dd74 100644
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@@ -816,8 +816,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[0][0][0][0])
         
         return x