Update davit.py

3 years ago · 2a55d9516e
parent 3a422f735a
commit 2a55d9516e
1 changed files with 0 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -542,7 +542,6 @@ def checkpoint_filter_fn(state_dict, model):
        state_dict = state_dict['state_dict']
    out_dict = {}
    import re
    for k, v in state_dict.items():
        k = k.replace('head.', 'head.fc.')