mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2024-09-22 10:14:35 -07:00
commit a22ce0a329
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
timm/models/mobilenetv3.py
View File

@ -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 """