From e6d7df40ecb9595b9e4f2f5b69f2633f81cee9ce Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 24 Jun 2022 21:32:44 -0700 Subject: [PATCH] no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg --- timm/models/helpers.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/timm/models/helpers.py b/timm/models/helpers.py index 11630bb6..fda84171 100644 --- a/timm/models/helpers.py +++ b/timm/models/helpers.py @@ -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)