Add lcnet_100 and mnasnet_small weights

pull/659/merge
Ross Wightman 3 years ago
parent 5ccf682a8f
commit 4df51f3932

@ -23,6 +23,13 @@ I'm fortunate to be able to dedicate significant time and money of my own suppor
## What's New
### Jan 6, 2022
* Tried training a few smaller models, two are good so far, more on the way...
* `mnasnet_small` - 65.6 top-1
* `lcnet_100` - 72.1 top-1
* TinyNet models added by [rsomani95](https://github.com/rsomani95)
* LCNet added via MobileNetV3 architecture
### Nov 22, 2021
* A number of updated weights anew new model defs
* `eca_halonext26ts` - 79.5 @ 256

@ -350,6 +350,7 @@ if 'GITHUB_ACTIONS' in os.environ:
'vit_large*',
'vit_base_patch8*',
'xcit_large*',
'*evob', '*evos', # until norm_norm_norm branch is merged
]
@ -427,7 +428,6 @@ if 'GITHUB_ACTIONS' not in os.environ:
'deit_*_distilled_patch16_224',
'levit*',
'pit_*_distilled_224',
'*evob', '*evos', # until norm_norm_norm branch is merged
] + EXCLUDE_FX_FILTERS

@ -77,7 +77,8 @@ default_cfgs = {
'semnasnet_100': _cfg(
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=''),
'mnasnet_small': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mnasnet_small-eb785140.pth'),
'mobilenetv2_035': _cfg(
url=''),

@ -82,7 +82,10 @@ default_cfgs = {
"lcnet_035": _cfg(),
"lcnet_050": _cfg(),
"lcnet_075": _cfg(),
"lcnet_100": _cfg(),
"lcnet_100": _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/lcnet_100-a929038c.pth',
interpolation='bicubic',
),
"lcnet_150": _cfg(),
}

Loading…
Cancel
Save