From 6e5be758dc29d0677802b03d70cfe2609ced8ce8 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Tue, 6 Dec 2022 17:10:37 -0800
Subject: [PATCH] Update davit.py

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

diff --git a/timm/models/davit.py b/timm/models/davit.py
index c48fb864..444f21f3 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -546,7 +546,8 @@ def checkpoint_filter_fn(state_dict, model):
     out_dict = {}
     import re
     for k, v in state_dict.items():
-        k = k.replace('norms.', 'head.norm.')
+    
+        k = k.replace('head.', 'head.fc.')
         out_dict[k] = v
     return out_dict