Skip to content

Commit 83b40c5

Browse files
committed
Last batch of small model weights (for now). mobilenetv3_small 050/075/100 and updated mnasnet_small with lambc/lamb optimizer.
1 parent 7f73252 commit 83b40c5

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

timm/models/efficientnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def _cfg(url='', **kwargs):
7979
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'),
8080
'semnasnet_140': _cfg(url=''),
8181
'mnasnet_small': _cfg(
82-
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_small-eb785140.pth'),
82+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_small_lamb-aff75073.pth'),
8383

8484
'mobilenetv2_035': _cfg(
8585
url=''),

timm/models/mobilenetv3.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,15 @@ def _cfg(url='', **kwargs):
4747
interpolation='bilinear', mean=(0, 0, 0), std=(1, 1, 1),
4848
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),
4949

50-
'mobilenetv3_small_050': _cfg(url=''),
51-
'mobilenetv3_small_075': _cfg(url=''),
52-
'mobilenetv3_small_100': _cfg(url=''),
50+
'mobilenetv3_small_050': _cfg(
51+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_050_lambc-4b7bbe87.pth',
52+
interpolation='bicubic'),
53+
'mobilenetv3_small_075': _cfg(
54+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_075_lambc-384766db.pth',
55+
interpolation='bicubic'),
56+
'mobilenetv3_small_100': _cfg(
57+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_100_lamb-266a294c.pth',
58+
interpolation='bicubic'),
5359

5460
'mobilenetv3_rw': _cfg(
5561
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_100-35495452.pth',

0 commit comments

Comments
 (0)