Pretrained cfg name mismatch

This commit is contained in:
Ross Wightman 2024-06-11 21:16:54 -07:00
parent 88efca1be2
commit c63da1405c

View File

@ -1256,7 +1256,7 @@ def mobilenetv4_hybrid_medium_075(pretrained: bool = False, **kwargs) -> MobileN
@register_model
def mobilenetv4_hybrid_large_075(pretrained: bool = False, **kwargs) -> MobileNetV3:
""" MobileNet V4 Hybrid"""
model = _gen_mobilenet_v4('mobilenetv4_hybrid_large', 0.75, pretrained=pretrained, **kwargs)
model = _gen_mobilenet_v4('mobilenetv4_hybrid_large_075', 0.75, pretrained=pretrained, **kwargs)
return model