Skip to content

Commit c1cb564

Browse files
committed
Add weight fore mobilenetv4 small 0.5, change 0.25 -> 0.35
1 parent a22ce0a commit c1cb564

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

timm/models/mobilenetv3.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1011,6 +1011,13 @@ def _cfg(url: str = '', **kwargs):
10111011
),
10121012
"lcnet_150.untrained": _cfg(),
10131013

1014+
'mobilenetv4_conv_small_035.untrained': _cfg(
1015+
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
1016+
test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'),
1017+
'mobilenetv4_conv_small_050.e3000_r224_in1k': _cfg(
1018+
hf_hub_id='timm/',
1019+
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
1020+
test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'),
10141021
'mobilenetv4_conv_small.e2400_r224_in1k': _cfg(
10151022
hf_hub_id='timm/',
10161023
test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'),
@@ -1254,20 +1261,19 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3:
12541261

12551262

12561263
@register_model
1257-
def mobilenetv4_conv_small_025(pretrained: bool = False, **kwargs) -> MobileNetV3:
1264+
def mobilenetv4_conv_small_035(pretrained: bool = False, **kwargs) -> MobileNetV3:
12581265
""" MobileNet V4 """
1259-
model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.25, pretrained=pretrained, **kwargs)
1266+
model = _gen_mobilenet_v4('mobilenetv4_conv_small_035', 0.35, pretrained=pretrained, **kwargs)
12601267
return model
12611268

12621269

12631270
@register_model
12641271
def mobilenetv4_conv_small_050(pretrained: bool = False, **kwargs) -> MobileNetV3:
12651272
""" MobileNet V4 """
1266-
model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.50, pretrained=pretrained, **kwargs)
1273+
model = _gen_mobilenet_v4('mobilenetv4_conv_small_050', 0.50, pretrained=pretrained, **kwargs)
12671274
return model
12681275

12691276

1270-
12711277
@register_model
12721278
def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3:
12731279
""" MobileNet V4 """

0 commit comments

Comments
 (0)