Commit Graph

131 Commits (51c432150a0019f6193ca53b1e0fc21b86dc2c2a)

Author SHA1 Message Date
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.
4 years ago
Ross Wightman 39f27c1add Almost ready to merge, vovnet typo, version bump, readme addition
4 years ago
Ross Wightman 208e7912f7 Missed one of the abalation model entrypoints, update README
4 years ago
Ross Wightman f4cdc2ac31 Add ResNeSt models
4 years ago
Ross Wightman 353a79aeba
Update README.md
4 years ago
Ross Wightman a4d20a1cb8
Update README.md
4 years ago
Ross Wightman c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
5 years ago
Ross Wightman e34074b4da Add final weights for MobileNet-V2 experiments
5 years ago
Ross Wightman 5a16c533ff Add better resnext50_32x4d weights trained by andravin
5 years ago
Ross Wightman 3406e582cf Add EfficientNet-Lite results, update README
5 years ago
Ross Wightman c16f25ced2 Add MobileNetV3 Large weights, results, update README and sotabench for merge
5 years ago
Ross Wightman f1860ef3a5 Update README.md, check in hubconf experiment (fully dynamic)
5 years ago
Ross Wightman b9496995d8 Add L2-475 PyTorch preprocessing result, update sotabench for new weights
5 years ago
Ross Wightman ba15ca47e8 Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
5 years ago
Ross Wightman 5eb0e363a6
Update README.md
5 years ago
Ross Wightman 5c4991a088 Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
5 years ago
Ross Wightman 4808b3c32f Bump version for PyPi update, fix few out of date README items/mistakes, add README updates for TF EfficientNet-B8 (RandAugment)
5 years ago
Ross Wightman b18c19901e
Update README.md
5 years ago
Ross Wightman 12dbc74742 New ResNet50 JSD + RandAugment weights
5 years ago
Ross Wightman e39aae56b4
Update README.md
5 years ago
Ross Wightman 7a17ee9c3b Update README.md
5 years ago
Ross Wightman 2a88412413
Update README.md
5 years ago
Ross Wightman cfa951bceb
Update README.md
5 years ago
Ross Wightman 7622015258
Update README.md
5 years ago
Ross Wightman ec0dd4053a Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
5 years ago
Ross Wightman a7fe891a13
Update README.md
5 years ago
Ross Wightman 0554b7942c
Update README.md
5 years ago
Ross Wightman 53f578e16f Update README with B3 training details
5 years ago
Ross Wightman acc3ed2b8c Add EfficientNet-B3 weights, trained from scratch with RA.
5 years ago
rwightman 1f4498f217 Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d
5 years ago
Ross Wightman 73b78459dc Add update RandAugment MixNet-XL weights
5 years ago
Ross Wightman 5d7af97981
Update README.md
5 years ago
Ross Wightman 3129bdb2c1
Update README.md
5 years ago
Ross Wightman ff421e5e09 New PyTorch trained EfficientNet-B2 weights with my RandAugment impl
5 years ago
Ross Wightman 5259dbcbb4 Update README.md with latest changes
5 years ago
Ross Wightman 31453b039e Update Auto/RandAugment comments, README, more.
5 years ago
Ross Wightman 0d58c50fb1 Add TF RandAug weights for B5/B7 EfficientNet models.
5 years ago
Ross Wightman 2680ad14bb Add Res2Net and DLA to README
5 years ago
Ross Wightman fac58f609a Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.
5 years ago
Ross Wightman 73fbd97ed4 Add weights for my MixNet-XL creation, include README updates for EdgeTPU models
5 years ago
Ross Wightman 3b4868f6dc A few more additions to Gluon Xception models to match interface of others.
5 years ago
Ross Wightman 9b35818811
Update README.md
5 years ago
Ross Wightman 4fe2da558c Add MixNet Small and Large PyTorch native weights (no same padding)
5 years ago
Ross Wightman 15e66863b1
Update README.md
5 years ago
Ross Wightman e879cf52fa Update validation scores for new TF EfficientNet weights.
5 years ago
Ross Wightman b35050a144
Update README.md
5 years ago
Ross Wightman 857f33015a Add native PyTorch weights for MixNet-Medium with no SAME padding necessary. Remove unused block of code.
5 years ago
Ross Wightman dfa9298b4e Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
5 years ago
Ross Wightman d97510fd4a Add new results csv and update README with 3 new ResNet weight results
5 years ago
Ross Wightman 5684c6af32
Update README.md
5 years ago
Ross Wightman a6b2f6eca5 Update README, bump version
5 years ago
Ross Wightman c3287aafb3 Slight improvement in EfficientNet-B2 native PyTorch weights
5 years ago
Ross Wightman b8762cc67d Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.
5 years ago
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...
5 years ago
Ross Wightman 8512436436 Add instagram pretrained ResNeXt models from https://pytorch.org/hub/facebookresearch_WSL-Images_resnext/, update README
5 years ago
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
5 years ago
Ross Wightman 1019414fd2 Add ported Tensorflow EfficientNet B4/B5 weights
5 years ago
Ross Wightman c6b8259ad6
Update README.md
5 years ago
Ross Wightman f3134973b5 Add the MnasNet-B1 variant weights, add A1/B1 model names as in the stand-alone repo, remove a bit of unused code
5 years ago
Ross Wightman 2060e433c0 Add native PyTorch weights for SE-MnasNet aka MnasNet-A1
5 years ago
Ross Wightman 7d17394bdc Add native PyTorch weights for EfficientNet-B0 w/ top-1 > 76.9
5 years ago
Ross Wightman 105d5702d7
Update README.md
5 years ago
Ross Wightman 6688d0669f PyTorch trained MobileNetV3 weights that match/best paper. Update fbnetc weights to improved copy.
5 years ago
Zhun Zhong b32361c9f8
Correct name
5 years ago
Ross Wightman 99122aac1c Replace ResNet-34 default weights with a great result from my experiments.
5 years ago
Ross Wightman 4bb5e9b224 Ported Tensorflow pretrained EfficientNet weights and some model cleanup
5 years ago
Ross Wightman 7a6d61566e Add EfficientNet impl, change existing depth_multipler -> channel_multiplier as definitions have been confused
5 years ago
Ross Wightman 99ab1b1276
Update README.md
5 years ago
Ross Wightman c2de3a922b
Update README.md
5 years ago
Ross Wightman 2da0b4dbc1 Add inception_v3 models via torchvision, 4 different pretrained weight choices
5 years ago
Ross Wightman af1a68d2e1
Update README.md
5 years ago
Ross Wightman a4516fe0fb Add pretrained weights for FBNet-C
5 years ago
Ross Wightman a72e9b67ee
Update README.md
5 years ago
Ross Wightman e3377b0409 Add some info to README
5 years ago
Ross Wightman fee607edf6 Mixup implemention in progress
5 years ago
Ross Wightman db056d97e2 Add MobileNetV3 and associated changes hard-swish, hard-sigmoid, efficient head, etc
5 years ago
Ross Wightman 0956cd4b66
Update README.md
5 years ago
Ross Wightman 1cf3ea0467
Update README.md
6 years ago
Ross Wightman 073d31a076
Update README.md
6 years ago
Ross Wightman 7ba78aaaeb
Update README.md
6 years ago
Ross Wightman e8e8bce335
Create README.md
6 years ago