From 9ca343717826578b0e003e78b694361621c2b0ef Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 14 Jan 2022 13:13:10 -0800 Subject: [PATCH] Add some more small model weights lcnet, mnas, mnv2 --- timm/models/efficientnet.py | 7 +++++-- timm/models/mobilenetv3.py | 10 ++++++++-- 2 files changed, 13 insertions(+), 4 deletions(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 87e033e0..9c0d5af6 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -73,7 +73,8 @@ default_cfgs = { 'mnasnet_140': _cfg(url=''), 'semnasnet_050': _cfg(url=''), - 'semnasnet_075': _cfg(url=''), + 'semnasnet_075': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/semnasnet_075-18710866.pth'), 'semnasnet_100': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'), 'semnasnet_140': _cfg(url=''), @@ -83,7 +84,9 @@ default_cfgs = { 'mobilenetv2_035': _cfg( url=''), 'mobilenetv2_050': _cfg( - url=''), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv2_050-3d30d450.pth', + interpolation='bicubic', + ), 'mobilenetv2_075': _cfg( url=''), 'mobilenetv2_100': _cfg( diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 491de284..4ab4568d 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -85,8 +85,14 @@ default_cfgs = { input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95), "lcnet_035": _cfg(), - "lcnet_050": _cfg(), - "lcnet_075": _cfg(), + "lcnet_050": _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_050-f447553b.pth', + interpolation='bicubic', + ), + "lcnet_075": _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_075-318cad2c.pth', + interpolation='bicubic', + ), "lcnet_100": _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth', interpolation='bicubic',