Skip to content

Commit 9ca3437

Browse files
committed
Add some more small model weights lcnet, mnas, mnv2
1 parent fa6463c commit 9ca3437

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

timm/models/efficientnet.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,8 @@ def _cfg(url='', **kwargs):
7373
'mnasnet_140': _cfg(url=''),
7474

7575
'semnasnet_050': _cfg(url=''),
76-
'semnasnet_075': _cfg(url=''),
76+
'semnasnet_075': _cfg(
77+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/semnasnet_075-18710866.pth'),
7778
'semnasnet_100': _cfg(
7879
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'),
7980
'semnasnet_140': _cfg(url=''),
@@ -83,7 +84,9 @@ def _cfg(url='', **kwargs):
8384
'mobilenetv2_035': _cfg(
8485
url=''),
8586
'mobilenetv2_050': _cfg(
86-
url=''),
87+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv2_050-3d30d450.pth',
88+
interpolation='bicubic',
89+
),
8790
'mobilenetv2_075': _cfg(
8891
url=''),
8992
'mobilenetv2_100': _cfg(

timm/models/mobilenetv3.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,14 @@ def _cfg(url='', **kwargs):
8585
input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95),
8686

8787
"lcnet_035": _cfg(),
88-
"lcnet_050": _cfg(),
89-
"lcnet_075": _cfg(),
88+
"lcnet_050": _cfg(
89+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_050-f447553b.pth',
90+
interpolation='bicubic',
91+
),
92+
"lcnet_075": _cfg(
93+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_075-318cad2c.pth',
94+
interpolation='bicubic',
95+
),
9096
"lcnet_100": _cfg(
9197
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth',
9298
interpolation='bicubic',

0 commit comments

Comments
 (0)