diff --git a/timm/models/hrnet.py b/timm/models/hrnet.py index e56a550936..20ea7674be 100644 --- a/timm/models/hrnet.py +++ b/timm/models/hrnet.py @@ -889,7 +889,9 @@ def _cfg(url='', **kwargs): default_cfgs = generate_default_cfgs({ + 'hrnet_w18_small.gluon_in1k': _cfg(hf_hub_id='timm/', interpolation='bicubic'), 'hrnet_w18_small.ms_in1k': _cfg(hf_hub_id='timm/'), + 'hrnet_w18_small_v2.gluon_in1k': _cfg(hf_hub_id='timm/', interpolation='bicubic'), 'hrnet_w18_small_v2.ms_in1k': _cfg(hf_hub_id='timm/'), 'hrnet_w18.ms_aug_in1k': _cfg( hf_hub_id='timm/',