Skip to content

Commit d138a9b

Browse files
committed
Add gluon hrnet small weights, fix #1895
1 parent 76d1669 commit d138a9b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

timm/models/hrnet.py

+2
Original file line numberDiff line numberDiff line change
@@ -889,7 +889,9 @@ def _cfg(url='', **kwargs):
889889

890890

891891
default_cfgs = generate_default_cfgs({
892+
'hrnet_w18_small.gluon_in1k': _cfg(hf_hub_id='timm/', interpolation='bicubic'),
892893
'hrnet_w18_small.ms_in1k': _cfg(hf_hub_id='timm/'),
894+
'hrnet_w18_small_v2.gluon_in1k': _cfg(hf_hub_id='timm/', interpolation='bicubic'),
893895
'hrnet_w18_small_v2.ms_in1k': _cfg(hf_hub_id='timm/'),
894896
'hrnet_w18.ms_aug_in1k': _cfg(
895897
hf_hub_id='timm/',

0 commit comments

Comments
 (0)