diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 6f09dd74..a108afda 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -816,10 +816,8 @@ class MetaFormer(nn.Module): def forward_features(self, x): x = self.patch_embed(x) - #x = self.stages(x) - for i, stage in enumerate(self.stages): - x = stage(x) - print(x[0][0][0][0]) + x = self.stages(x) + return x