Skip to content

Commit 4aa166d

Browse files
committed
Add hgnet ssld weights
1 parent 159e916 commit 4aa166d

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

timm/models/hgnet.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -612,11 +612,18 @@ def _cfg(url='', **kwargs):
612612
'hgnet_tiny.paddle_in1k': _cfg(
613613
first_conv='stem.0.conv',
614614
hf_hub_id='timm/'),
615+
'hgnet_tiny.ssld_in1k': _cfg(
616+
first_conv='stem.0.conv',
617+
hf_hub_id='timm/'),
615618
'hgnet_small.paddle_in1k': _cfg(
616619
first_conv='stem.0.conv',
617620
hf_hub_id='timm/'),
618-
'hgnet_base': _cfg(
619-
first_conv='stem.0.conv'),
621+
'hgnet_small.ssld_in1k': _cfg(
622+
first_conv='stem.0.conv',
623+
hf_hub_id='timm/'),
624+
'hgnet_base.ssld_in1k': _cfg(
625+
first_conv='stem.0.conv',
626+
hf_hub_id='timm/'),
620627
'hgnetv2_b0.ssld_in1k': _cfg(
621628
first_conv='stem.stem1.conv',
622629
hf_hub_id='timm/'),

0 commit comments

Comments
 (0)