From a97963bff1f5543e5f95e57001c2acf4ec5a6f9f Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Fri, 16 Dec 2022 02:40:00 -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 a51c8243..3ee6c4e6 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -569,6 +569,7 @@ def checkpoint_filter_fn(state_dict, model): import re out_dict = {} for k, v in state_dict.items(): + k = k.replace('stages.0.patch_embed', 'patch_embed') 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('head.', 'head.fc.')