Change order of checkpoitn filtering fn application in builder, try dict, model variant first

pull/1641/head
Ross Wightman 2023-01-20 14:48:54 -08:00
parent e9f1376cde
commit 32f252381d
1 changed files with 3 additions and 3 deletions

View File

@ -179,11 +179,11 @@ def load_pretrained(
return
if filter_fn is not None:
# for backwards compat with filter fn that take one arg, try one first, the two
try:
state_dict = filter_fn(state_dict)
except TypeError:
state_dict = filter_fn(state_dict, model)
except TypeError as e:
# for backwards compat with filter fn that take one arg
state_dict = filter_fn(state_dict)
input_convs = pretrained_cfg.get('first_conv', None)
if input_convs is not None and in_chans != 3: