pull/595/head
talrid 4 years ago
parent 19e1b67a84
commit 8c1f03e56c

@ -375,14 +375,18 @@ def mobilenetv3_large_100(pretrained=False, **kwargs):
@register_model
def mobilenetv3_large_100_1k_miil(pretrained=False, **kwargs):
""" MobileNet V3 """
""" MobileNet V3
Weights taken from: https://github.com/Alibaba-MIIL/ImageNet21K
"""
model = _gen_mobilenet_v3('mobilenetv3_large_100_1k_miil_78_0', 1.0, pretrained=pretrained, **kwargs)
return model
@register_model
def mobilenetv3_large_100_in21k_miil(pretrained=False, **kwargs):
""" MobileNet V3 """
""" MobileNet V3, 21k pretraining
Weights taken from: https://github.com/Alibaba-MIIL/ImageNet21K
"""
model = _gen_mobilenet_v3('mobilenetv3_large_100_in21k_miil', 1.0, pretrained=pretrained, **kwargs)
return model

Loading…
Cancel
Save