comment

Author: talrid

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