From e060223cee7ddd5a6e9c88904e1bce49c2842732 Mon Sep 17 00:00:00 2001 From: Rahul Somani Date: Sat, 1 May 2021 12:10:49 +0530 Subject: [PATCH] add default configs for aa models --- timm/models/mobilenetv3.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index ce59c5b5..d9b4d755 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -39,6 +39,8 @@ default_cfgs = { 'mobilenetv3_large_100': _cfg( interpolation='bicubic', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_large_100_ra-f55367f5.pth'), + 'mobilenetv3_large_075_aa': _cfg(url=''), + 'mobilenetv3_large_100_aa': _cfg(url=''), 'mobilenetv3_large_100_miil': _cfg( interpolation='bilinear', mean=(0, 0, 0), std=(1, 1, 1), url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm/mobilenetv3_large_100_1k_miil_78_0.pth'), @@ -415,7 +417,7 @@ def mobilenetv3_large_075_aa(pretrained=False, aa_layer=BlurPool2d, **kwargs): @register_model def mobilenetv3_large_100_aa(pretrained=False, aa_layer=BlurPool2d, **kwargs): """ MobileNet V3 """ - model = _gen_mobilenet_v3('mobilenetv3_large_100', 1.0, pretrained=pretrained, aa_layer=aa_layer, **kwargs) + model = _gen_mobilenet_v3('mobilenetv3_large_100_aa', 1.0, pretrained=pretrained, aa_layer=aa_layer, **kwargs) return model