From e8fb866ccf39a0c2922649771ec3fa54c3a9655a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=96=B9=E6=9B=A6?= <xi.fang@smartmore.com>
Date: Wed, 2 Aug 2023 14:40:01 +0800
Subject: [PATCH] fix efficientvit_msra pool

---
 timm/models/efficientvit_msra.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/efficientvit_msra.py b/timm/models/efficientvit_msra.py
index 00bfc055..bef462e0 100644
--- a/timm/models/efficientvit_msra.py
+++ b/timm/models/efficientvit_msra.py
@@ -384,6 +384,7 @@ class EfficientViTMSRA(nn.Module):
         return x
 
     def forward_head(self, x, pre_logits: bool = False):
+        x = self.global_pool(x)
         return x if pre_logits else self.head(x)
 
     def forward(self, x):