Commit Graph

175 Commits (c1b3cea19df84beb9d5e141272b383e9ba9a0980)

Author SHA1 Message Date
Ross Wightman 5a196dddf6 Update README.md with latest, bump version to 0.4.8 2021-04-12 13:15:00 -07:00
Ross Wightman acbd698c83 Update README.md with updates. Small tweak to head_dist handling. 2021-04-01 17:49:05 -07:00
Ross Wightman 740f32c96a Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6 2021-03-17 13:55:32 -07:00
Ross Wightman 5c7d298234 Update README.md, documentation, add TOC, sponsors, and links to timmdocs and paperswithcode 2021-03-10 10:24:23 -08:00
Ross Wightman f1f332fee5
Update README.md 2021-03-07 16:43:52 -08:00
Ross Wightman f57db99101 Update README, fix iabn pip version print. 2021-03-07 16:17:06 -08:00
Ross Wightman 678ba4e0a2 Add NFNet-F model weights ported from DeepMind Haiku impl and new set of models w/ compatible config. 2021-02-18 12:28:46 -08:00
Ross Wightman 361fd0fc40
Update README.md 2021-02-16 10:27:41 -08:00
Ross Wightman 9de2ec5e44 Update README for AGC and bump version to 0.4.4 2021-02-16 09:13:03 -08:00
Ross Wightman d86dbe45c2 Update README.md and few more comments 2021-02-12 22:07:18 -08:00
Ross Wightman e4de077021 Add first 'Normalizer Free' models. nf_regnet_b1 79.3 @ 288x288 test, and nf_resnet50 80.3 @ 256x256 test (80.68 @ 288x288). 2021-02-11 13:20:11 -08:00
Ross Wightman ca9b078ac7 Update README.md and docs. Version bumped to 0.4.3 2021-02-10 14:46:07 -08:00
Ross Wightman aaa715b1e9
Update README.md 2021-02-08 12:04:48 -08:00
Ross Wightman a39c3ee216
Merge branch 'master' into eca-weights 2021-02-08 11:52:31 -08:00
Ross Wightman e9d6fe293c Update README for new weights. Version 0.4.2 2021-02-08 11:51:16 -08:00
Ross Wightman 3a7aa95f7e
Update README.md 2021-02-05 07:39:21 -08:00
Ross Wightman 2c988c3b6e Update README.md for NF-nets, bump version to 0.4.1 for merge 2021-01-30 23:19:45 -08:00
Ross Wightman 745bc5f723 Mention 21k ViT weights in README.md 2021-01-25 12:26:55 -08:00
Ross Wightman 587780e56b Update README.md and bump version to 0.4.0 2021-01-25 11:22:11 -08:00
Ross Wightman 58ccf43150 Add BiT references and knowledge distill links to readme/docs 2021-01-16 12:30:02 -08:00
Ross Wightman 19816fe226 Add citation info 2021-01-03 15:46:12 -08:00
Ross Wightman 1d01c2b68c
Update README.md 2021-01-03 12:49:31 -08:00
Ross Wightman 7a75b8d033
Update README.md 2020-12-18 17:27:57 -08:00
Ross Wightman b1f1228a41 Add ResNet101D, 152D, and 200D weights, remove meh 66d model 2020-12-18 17:13:37 -08:00
Ross Wightman 51d74d91da Update README.md 2020-12-07 09:59:50 -08:00
Roger Shieh a7f6126b92
Update README.md 2020-11-24 09:02:37 +08:00
Ross Wightman 46f15443be Update README.md and docs in prep for 0.3.0 PyPi release. 2020-10-30 13:27:19 -07:00
Ross Wightman 736f209e7d Update vision transformers to be compatible with official code. Port official ViT weights from jax impl. 2020-10-26 18:42:11 -07:00
Ross Wightman 27a93e9de7 Improve test crop for ViT models. Small now 77.85, added base weights at 79.35 top-1. 2020-10-21 23:35:25 -07:00
Ross Wightman be53107e8a Update README, ensure vit excluded from all tests (not ready) 2020-10-13 14:51:08 -07:00
Ross Wightman c40384f5bd Add ResNet weights. 80.5 (top-1) ResNet-50-D, 77.1 ResNet-34-D, 72.7 ResNet-18-D. 2020-09-18 12:05:37 -07:00
Ross Wightman 9ce42d5c5a
Update README.md 2020-09-03 10:48:25 -07:00
Ross Wightman 0729dbe865
Update README.md 2020-09-03 10:48:11 -07:00
Ross Wightman 33f8a1bf36 Updated README, add wide_resnet50_2 and seresnext50_32x4d weights 2020-09-03 10:45:17 -07:00
Ross Wightman 6d158adfe9
Update README.md 2020-08-26 15:15:51 -07:00
Ross Wightman fa26f6c1cf
Update README.md 2020-08-12 23:26:59 -07:00
Ross Wightman f614df7921 Bump version to 0.2.1 and update README 2020-08-12 18:05:07 -07:00
Ross Wightman 6e9d6172c8 A few typos and missed updates in changelog 2020-08-05 22:19:59 -07:00
Ross Wightman 1696499ce5 Bump version to 0.2.0, ready to roll (I think) 2020-08-05 16:55:18 -07:00
Ross Wightman e62758cf4f More documentation updates, fix a typo 2020-08-05 15:59:31 -07:00
Ross Wightman 5e333b899f README / doc tweaks 2020-08-05 13:17:39 -07:00
Ross Wightman e3f58fc90c Continuing README and documentation updates 2020-08-05 10:38:35 -07:00
Ross Wightman e3c11a36dc Start updating README and docs 2020-08-04 16:55:02 -07:00
Ross Wightman 328339a086 Add ESE-VoVNet-19-DW weights 2020-06-15 14:31:55 -07:00
Ross Wightman f225ae8e59 Update README with model results and attribution. Make scheduler factory bit more robust to arg differences, add noise to plateau lr and fix min/max. 2020-06-12 13:33:12 -07:00
Ross Wightman 39f27c1add Almost ready to merge, vovnet typo, version bump, readme addition 2020-06-11 15:41:58 -07:00
Ross Wightman 208e7912f7 Missed one of the abalation model entrypoints, update README 2020-05-12 13:36:31 -07:00
Ross Wightman f4cdc2ac31 Add ResNeSt models 2020-05-11 23:27:09 -07:00
Ross Wightman 353a79aeba
Update README.md 2020-05-03 21:24:49 -07:00
Ross Wightman a4d20a1cb8
Update README.md 2020-05-01 19:22:35 -07:00
Ross Wightman c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
MobileNet-V2 experiments
2020-04-05 17:32:06 -07:00
Ross Wightman e34074b4da Add final weights for MobileNet-V2 experiments 2020-04-05 17:31:02 -07:00
Ross Wightman 5a16c533ff Add better resnext50_32x4d weights trained by andravin 2020-03-18 14:43:50 -07:00
Ross Wightman 3406e582cf Add EfficientNet-Lite results, update README 2020-03-18 13:12:30 -07:00
Ross Wightman c16f25ced2 Add MobileNetV3 Large weights, results, update README and sotabench for merge 2020-02-29 20:37:20 -08:00
Ross Wightman f1860ef3a5 Update README.md, check in hubconf experiment (fully dynamic) 2020-02-18 14:41:14 -08:00
Ross Wightman b9496995d8 Add L2-475 PyTorch preprocessing result, update sotabench for new weights 2020-02-12 13:20:37 -08:00
Ross Wightman ba15ca47e8 Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU 2020-02-12 11:26:38 -08:00
Ross Wightman 5eb0e363a6
Update README.md 2020-02-06 12:56:46 -08:00
Ross Wightman 5c4991a088 Add PyTorch trained EfficientNet-ES weights from Andrew Lavin 2020-02-06 12:53:55 -08:00
Ross Wightman 4808b3c32f Bump version for PyPi update, fix few out of date README items/mistakes, add README updates for TF EfficientNet-B8 (RandAugment) 2020-02-03 11:44:17 -08:00
Ross Wightman b18c19901e
Update README.md
Typo
2020-01-31 10:59:31 -08:00
Ross Wightman 12dbc74742 New ResNet50 JSD + RandAugment weights 2020-01-31 10:55:54 -08:00
Ross Wightman e39aae56b4
Update README.md
Fix typo
2020-01-13 17:49:49 -08:00
Ross Wightman 7a17ee9c3b Update README.md 2020-01-12 18:16:36 -08:00
Ross Wightman 2a88412413
Update README.md 2020-01-11 12:08:15 -08:00
Ross Wightman cfa951bceb
Update README.md 2020-01-07 08:52:12 -08:00
Ross Wightman 7622015258
Update README.md 2020-01-04 11:51:38 -08:00
Ross Wightman ec0dd4053a Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko 2020-01-03 17:18:46 -08:00
Ross Wightman a7fe891a13
Update README.md
Update readme with SelecSLS details.
2019-12-30 16:32:49 -08:00
Ross Wightman 0554b7942c
Update README.md 2019-12-29 09:15:00 -08:00
Ross Wightman 53f578e16f Update README with B3 training details 2019-12-28 17:51:08 -08:00
Ross Wightman acc3ed2b8c Add EfficientNet-B3 weights, trained from scratch with RA. 2019-12-28 17:24:15 -08:00
rwightman 1f4498f217 Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d 2019-12-28 11:43:50 -08:00
Ross Wightman 73b78459dc Add update RandAugment MixNet-XL weights 2019-12-24 10:08:24 -08:00
Ross Wightman 5d7af97981
Update README.md
Update latest training hparam/command line with correct RandAugment config string (`noise` -> `mstd`)
2019-12-05 11:47:57 -08:00
Ross Wightman 3129bdb2c1
Update README.md 2019-12-04 11:13:25 -08:00
Ross Wightman ff421e5e09 New PyTorch trained EfficientNet-B2 weights with my RandAugment impl 2019-12-04 11:09:47 -08:00
Ross Wightman 5259dbcbb4 Update README.md with latest changes 2019-11-29 17:58:33 -08:00
Ross Wightman 31453b039e Update Auto/RandAugment comments, README, more.
* Add a weighted choice option for RandAugment
* Adjust magnitude noise/std naming, config
2019-11-22 13:24:52 -08:00
Ross Wightman 0d58c50fb1 Add TF RandAug weights for B5/B7 EfficientNet models. 2019-10-30 16:49:17 -07:00
Ross Wightman 2680ad14bb Add Res2Net and DLA to README 2019-09-04 17:38:59 -07:00
Ross Wightman fac58f609a Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.
* Add some of the trendy new optimizers. Decent results but not clearly better than the standards.
* Can create a None scheduler for constant LR
* ResNet defaults to zero_init of last BN in residual
* add resnet50d config
2019-08-28 00:14:10 -07:00
Ross Wightman 73fbd97ed4 Add weights for my MixNet-XL creation, include README updates for EdgeTPU models 2019-08-24 19:57:42 -07:00
Ross Wightman 3b4868f6dc A few more additions to Gluon Xception models to match interface of others. 2019-08-11 23:06:23 -07:00
Ross Wightman 9b35818811
Update README.md 2019-08-03 09:32:08 -07:00
Ross Wightman 4fe2da558c Add MixNet Small and Large PyTorch native weights (no same padding) 2019-08-02 23:22:48 -07:00
Ross Wightman 15e66863b1
Update README.md 2019-07-31 14:41:31 -07:00
Ross Wightman e879cf52fa Update validation scores for new TF EfficientNet weights. 2019-07-31 14:38:55 -07:00
Ross Wightman b35050a144
Update README.md 2019-07-30 18:33:40 -07:00
Ross Wightman 857f33015a Add native PyTorch weights for MixNet-Medium with no SAME padding necessary. Remove unused block of code. 2019-07-29 11:59:15 -07:00
Ross Wightman dfa9298b4e Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
* refactor 'same' convolution and add helper to use MixedConv2d when needed
* improve performance of 'same' padding for cases that can be handled statically
* add support for extra exp, pw, and dw kernel specs with grouping support to decoder/string defs for MixNet
* shuffle some args for a bit more consistency, a little less clutter overall in gen_efficientnet.py
2019-07-25 11:42:01 -07:00
Ross Wightman d97510fd4a Add new results csv and update README with 3 new ResNet weight results 2019-07-20 08:45:05 -07:00
Ross Wightman 5684c6af32
Update README.md
Add blurb about results csv
2019-07-04 10:52:03 -07:00
Ross Wightman a6b2f6eca5 Update README, bump version 2019-07-03 22:48:33 -07:00
Ross Wightman c3287aafb3 Slight improvement in EfficientNet-B2 native PyTorch weights 2019-06-29 16:17:29 -07:00
Ross Wightman b8762cc67d Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.
* Remove some models that don't exist as pretrained an likely never will (se)resnext152
* Add some torchvision weights as tv_ for models that I have added better weights for
* Add wide resnet recently added to torchvision along with resnext101-32x8d
* Add functionality to model registry to allow filtering on pretrained weight presence
2019-06-29 15:50:33 -07:00
Ross Wightman 13c19e213d Add native PyTorch EfficientNet B1 and B2 weights. Not quite where I want them, but hitting the brick wall and moving on to other projects... 2019-06-24 13:12:04 -07:00
Ross Wightman 8512436436 Add instagram pretrained ResNeXt models from https://pytorch.org/hub/facebookresearch_WSL-Images_resnext/, update README 2019-06-23 12:29:02 -07:00
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 2019-06-20 17:29:25 -07:00