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