From 0aa784027eaf3e56c9c7e40faa4e5b6df0d1fb4a Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sun, 8 Jan 2023 11:22:55 -0800
Subject: [PATCH] Update metaformers.py

---
 timm/models/metaformers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):