mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

comment

Browse Source
This commit is contained in:
talrid 2021-04-28 09:04:10 +03:00
parent 19e1b67a84
commit 8c1f03e56c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
timm/models/mobilenetv3.py
View File

@ -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