From 8c1f03e56cef6fceb1ad9307193ac4c5adf31014 Mon Sep 17 00:00:00 2001 From: talrid Date: Wed, 28 Apr 2021 09:04:10 +0300 Subject: [PATCH] comment --- timm/models/mobilenetv3.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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