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

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

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index 27964167..dc83c185 100644
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@@ -819,11 +819,13 @@ class MetaFormer(nn.Module):
         
     def forward_features(self, x):
         x = self.patch_embed(x)
+        print('timm')
         #x = self.stages(x)
+        '''
         for i, stage in enumerate(self.stages):
             x=stage(x)
             #print(x[0][0][0][0])
-        
+        '''
         
         return x