Commit Graph

193 Commits (3698e79ac5ea869e2e1ac9d22e8c84af8a99b57a)

Author SHA1 Message Date
Ross Wightman 392368e210 Add efficientnetv2_rw_t defs w/ weights, and gc variant, as well as gcresnet26ts for experiments. Version 0.4.13 2021-07-09 16:46:52 -07:00
Ross Wightman 6d8272e92c Add SAM pretrained model defs/weights for ViT B16 and B32 models. 2021-07-08 11:51:12 -07:00
Ross Wightman 81cd6863c8 Move aggregation (convpool) for nest into NestLevel, cleanup and enable features_only use. Finalize weight url. 2021-07-05 18:20:49 -07:00
Santiago Castro 49b38a51e3
Add color highlighting to BibTeX entry in README 2021-06-30 14:07:16 -07:00
Ross Wightman 02aaa785b9
Update README.md 2021-06-23 14:08:16 -07:00
Ross Wightman 20a2be14c3 Add gMLP-S weights, 79.6 top-1 2021-06-23 10:40:30 -07:00
Ross Wightman dc422820ec
Update README.md 2021-06-20 22:12:56 -07:00
Ross Wightman 9c9755a808 AugReg release 2021-06-20 17:46:06 -07:00
Ross Wightman 758c4438a7 Update README.md 2021-06-08 15:19:11 -07:00
Ross Wightman 307a935b79 Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA. 2021-05-31 13:18:11 -07:00
Ross Wightman 07d952c7a7
Merge pull request #637 from rwightman/levit_visformer_rednet
LeVit, Visformer, RedNet/Involution models and layers
2021-05-25 14:27:06 -07:00
Peter Vandenabeele 51c432150a README: fix simple typos 2021-05-25 22:43:54 +02:00
Ross Wightman 318360c3f9 Update README.md before merge. Bump version to 0.4.10 2021-05-25 12:26:16 -07:00
Ross Wightman 7077f16c6a Change 21k model naming from _21k to _in21k for consistency with existing 21k models. 2021-05-15 12:42:26 -07:00
Ross Wightman 2a72d38ba2
Update README.md 2021-05-14 23:06:49 -07:00
Ross Wightman 328249f11a Update README, tweak fine-tune effv2 model names. 2021-05-14 16:37:43 -07:00
Ross Wightman f480734ad5 Update README.md 2021-05-05 23:17:35 -07:00
Ross Wightman 858728799c Update README again. Add 101x3 BiT-M model to CI ignore since it's starting to fail in GitHub runners. 2021-04-13 13:36:24 -07:00
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