From 83b40c5a58b1fc43d053de537ef3201362cc4753 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 19 Jan 2022 10:01:18 -0800 Subject: [PATCH] Last batch of small model weights (for now). mobilenetv3_small 050/075/100 and updated mnasnet_small with lambc/lamb optimizer. --- timm/models/efficientnet.py | 2 +- timm/models/mobilenetv3.py | 12 +++++++++--- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 9c0d5af6..cb65ffbe 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -79,7 +79,7 @@ default_cfgs = { url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'), 'semnasnet_140': _cfg(url=''), 'mnasnet_small': _cfg( - url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_small-eb785140.pth'), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_small_lamb-aff75073.pth'), 'mobilenetv2_035': _cfg( url=''), diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 4ab4568d..80474122 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -47,9 +47,15 @@ default_cfgs = { 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_in21k_miil.pth', num_classes=11221), - 'mobilenetv3_small_050': _cfg(url=''), - 'mobilenetv3_small_075': _cfg(url=''), - 'mobilenetv3_small_100': _cfg(url=''), + 'mobilenetv3_small_050': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_050_lambc-4b7bbe87.pth', + interpolation='bicubic'), + 'mobilenetv3_small_075': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_075_lambc-384766db.pth', + interpolation='bicubic'), + 'mobilenetv3_small_100': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_100_lamb-266a294c.pth', + interpolation='bicubic'), 'mobilenetv3_rw': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_100-35495452.pth',