Add gluon hrnet small weights, fix #1895

huggingface · Aug 3, 2023 · d138a9b · d138a9b
1 parent 76d1669
commit d138a9b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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/',