mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2023-01-20 14:48:54 -08:00
parent e9f1376cde
commit 32f252381d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
timm/models/_builder.py
View File

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