Update metaformers.py

2 years ago · 0aa784027e
parent 1145720bec
commit 0aa784027e
1 changed files with 1 additions and 1 deletions
--- 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):