diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 0e9a58cb..fe5622c6 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -710,7 +710,7 @@ class MetaFormer(nn.Module): def forward(self, x): x = self.forward_features(x) - x = forward_head(x) + x = self.forward_head(x) return x def checkpoint_filter_fn(state_dict, model):