Add some more small model weights lcnet, mnas, mnv2

pull/1091/head
Ross Wightman 3 years ago committed by Ross Wightman
parent fa6463c936
commit 9ca3437178

@ -73,7 +73,8 @@ default_cfgs = {
'mnasnet_140': _cfg(url=''), 'mnasnet_140': _cfg(url=''),
'semnasnet_050': _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( 'semnasnet_100': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'), url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_a1-d9418771.pth'),
'semnasnet_140': _cfg(url=''), 'semnasnet_140': _cfg(url=''),
@ -83,7 +84,9 @@ default_cfgs = {
'mobilenetv2_035': _cfg( 'mobilenetv2_035': _cfg(
url=''), url=''),
'mobilenetv2_050': _cfg( '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( 'mobilenetv2_075': _cfg(
url=''), url=''),
'mobilenetv2_100': _cfg( 'mobilenetv2_100': _cfg(

@ -85,8 +85,14 @@ default_cfgs = {
input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95), input_size=(3, 240, 240), test_input_size=(3, 288, 288), crop_pct=0.95),
"lcnet_035": _cfg(), "lcnet_035": _cfg(),
"lcnet_050": _cfg(), "lcnet_050": _cfg(
"lcnet_075": _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( "lcnet_100": _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth',
interpolation='bicubic', interpolation='bicubic',

Loading…
Cancel
Save