Update davit.py

3 years ago · 6e5be758dc
parent 385a3bdb5e
commit 6e5be758dc
1 changed files with 2 additions and 1 deletions
--- 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