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.')