Update metaformers.py

2 years ago · 4ed934e000
parent 3f0b075367
commit 4ed934e000
1 changed files with 3 additions and 1 deletions
--- 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