From 2a55d9516efecb1dd2bb96a01e74dd6224882f66 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 6 Dec 2022 20:46:57 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 1 - 1 file changed, 1 deletion(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index ae7b3836..a307ba6a 100644 --- 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.')