Merge branch 'patch-1' of https://github.com/baorepo/pytorch-image-mo…

…dels into baorepo-patch-1
huggingface · Sep 22, 2024 · a22ce0a · a22ce0a
2 parents 9337463 + d6b8816
commit a22ce0a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git 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 """