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