Revert "Update metaformers.py"

This reverts commit 4ed934e000.
2 years ago · f985f35b91
parent a1882204ea
commit f985f35b91
1 changed files with 1 additions and 3 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -819,13 +819,11 @@ 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