diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 84bce1a5..42395388 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -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