diff --git a/README.md b/README.md index 619cffb4..a37d97eb 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/tests/test_models.py b/tests/test_models.py index 035db19e..73c3e0b6 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -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 diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 71443e4f..30d6fd6a 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -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=''), diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index f9fa1e4d..92a1f94a 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -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(), }