Skip to content

Commit e6d7df4

Browse files
committed
no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg
1 parent a29fba3 commit e6d7df4

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

timm/models/helpers.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -455,10 +455,9 @@ def update_pretrained_cfg_and_kwargs(pretrained_cfg, kwargs, kwargs_filter):
455455
filter_kwargs(kwargs, names=kwargs_filter)
456456

457457

458-
def resolve_pretrained_cfg(variant: str, **kwargs):
459-
pretrained_cfg = kwargs.pop('pretrained_cfg', None)
458+
def resolve_pretrained_cfg(variant: str, pretrained_cfg=None):
460459
if pretrained_cfg and isinstance(pretrained_cfg, dict):
461-
# highest priority, pretrained_cfg available and passed in args
460+
# highest priority, pretrained_cfg available and passed as arg
462461
return deepcopy(pretrained_cfg)
463462
# fallback to looking up pretrained cfg in model registry by variant identifier
464463
pretrained_cfg = get_pretrained_cfg(variant)

0 commit comments

Comments
 (0)