From 1470cebb6c6bce51dc41f20758ff2e1590cca614 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Fri, 16 Dec 2022 02:42:58 -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 3ee6c4e6..e44519df 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -569,9 +569,10 @@ 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('stages.0.patch_embed', 'patch_embed') k = k.replace('head.', 'head.fc.') k = k.replace('cpe.0', 'cpe1') k = k.replace('cpe.1', 'cpe2')