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

no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg

Browse Source
This commit is contained in:
Ross Wightman 2022-06-24 21:32:44 -07:00
parent a29fba307d
commit e6d7df40ec
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/helpers.py
View File

@ -455,10 +455,9 @@ def update_pretrained_cfg_and_kwargs(pretrained_cfg, kwargs, kwargs_filter):
filter_kwargs(kwargs, names=kwargs_filter)
def resolve_pretrained_cfg(variant: str, **kwargs):
pretrained_cfg = kwargs.pop('pretrained_cfg', None)
def resolve_pretrained_cfg(variant: str, pretrained_cfg=None):
if pretrained_cfg and isinstance(pretrained_cfg, dict):
# highest priority, pretrained_cfg available and passed in args
# highest priority, pretrained_cfg available and passed as arg
return deepcopy(pretrained_cfg)
# fallback to looking up pretrained cfg in model registry by variant identifier
pretrained_cfg = get_pretrained_cfg(variant)