From b1337d7f4c5d1c8fea012e221355bd9b4b175a28 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 7 Jan 2023 20:38:14 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/models/davit.py b/timm/models/davit.py index d0b5e44c..f21e799e 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -565,6 +565,7 @@ def checkpoint_filter_fn(state_dict, model): k = re.sub(r'patch_embeds.([0-9]+)', r'stages.\1.patch_embed', k) k = re.sub(r'main_blocks.([0-9]+)', r'stages.\1.blocks', k) k = k.replace('stages.0.patch_embed', 'stem') + k = k.replace('norms.', 'norm.') k = k.replace('head.', 'head.fc.') k = k.replace('cpe.0', 'cpe1') k = k.replace('cpe.1', 'cpe2')