diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 0aa0eebc5b..d430a9339a 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1253,6 +1253,21 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3: return model +@register_model +def mobilenetv4_conv_small_025(pretrained: bool = False, **kwargs) -> MobileNetV3: + """ MobileNet V4 """ + model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.25, pretrained=pretrained, **kwargs) + return model + + +@register_model +def mobilenetv4_conv_small_050(pretrained: bool = False, **kwargs) -> MobileNetV3: + """ MobileNet V4 """ + model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.50, pretrained=pretrained, **kwargs) + return model + + + @register_model def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 """