From 6b6510f30ab8d153f6fb47d1a3af6f21272d66f1 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Tue, 17 Jan 2023 11:49:10 -0800
Subject: [PATCH] Update metaformers.py

---
 timm/models/metaformers.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index 1356c769..5608b368 100644
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@@ -819,7 +819,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)
         
         
         return x