mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add lcnet_100 and mnasnet_small weights

Browse Source
This commit is contained in:
Ross Wightman 2022-01-06 22:21:05 -08:00
parent 5ccf682a8f
commit 4df51f3932
4 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
README.md
View File

@ -23,6 +23,13 @@ I'm fortunate to be able to dedicate significant time and money of my own suppor
## What's New ## 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 ### Nov 22, 2021
* A number of updated weights anew new model defs * A number of updated weights anew new model defs
* `eca_halonext26ts` - 79.5 @ 256 * `eca_halonext26ts` - 79.5 @ 256

2
tests/test_models.py
View File

@ -350,6 +350,7 @@ if 'GITHUB_ACTIONS' in os.environ:
'vit_large*', 'vit_large*',
'vit_base_patch8*', 'vit_base_patch8*',
'xcit_large*', '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', 'deit_*_distilled_patch16_224',
'levit*', 'levit*',
'pit_*_distilled_224', 'pit_*_distilled_224',
'*evob', '*evos', # until norm_norm_norm branch is merged
] + EXCLUDE_FX_FILTERS ] + EXCLUDE_FX_FILTERS

3
timm/models/efficientnet.py
View File

@ -77,7 +77,8 @@ default_cfgs = {
'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=''),
'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( 'mobilenetv2_035': _cfg(
url=''), url=''),

5
timm/models/mobilenetv3.py
View File

@ -82,7 +82,10 @@ default_cfgs = {
"lcnet_035": _cfg(), "lcnet_035": _cfg(),
"lcnet_050": _cfg(), "lcnet_050": _cfg(),
"lcnet_075": _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(), "lcnet_150": _cfg(),
} }