From 1d7c04e28af58c60fdb0317a86c3f5d5d0395283 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 02:42:10 -0800
Subject: [PATCH] Update davit.py

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

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 18898200..ee154d56 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -382,7 +382,7 @@ class DaViT(nn.Module):
         attn_drop_rate=0.,
         img_size=224,
         num_classes=1000,
-        global_pool='avg'#,
+        global_pool='avg',
         #features_only = False
     **kwargs):
         super().__init__()