Commit Graph

299 Commits (f098fda2ca48bf66b611237787b52b9614003a8b)
 

Author SHA1 Message Date
Ross Wightman f098fda2ca Add map_location='cpu' to ModelEma resume, should improve #72
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 d0eb59ef46 Remove unused default_init for EfficientNets, experimenting with fanout calc for #84
5 years ago
Ross Wightman cade829105 Add EfficientNet-ES to sotabench
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 d66819d1f3 Indentation mistake. Fixes #81
5 years ago
Ross Wightman 65cda1c63f
Merge pull request #83 from andravin/validation-batch-size-multiplier
5 years ago
Andrew Lavin b72013def8 Added commandline argument validation-batch-size-multiplier with default set to 1.
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 5c85389e26
Update README.md
5 years ago
Ross Wightman 820b73d280
Update README.md
5 years ago
Ross Wightman 82c0a2ff12
Update README.md
5 years ago
Ross Wightman 1ffd2d0694
Add results/README.md
5 years ago
Ross Wightman fd98fb33c5 Update sotabench with tf_efficientnet_b8 model
5 years ago
Ross Wightman 7c88356682 Add update results on ImageNet validation, ImageNetV2, ImageNet-A, and ImageNet-Sketch for all models
5 years ago
Ross Wightman 1daa303744 Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.
5 years ago
Ross Wightman 91534522f9 Add newly added TF ported EfficientNet-B8 weights (RandAugment)
5 years ago
Ross Wightman 82dd60b33c Rename results files for more clarity
5 years ago
Ross Wightman 5b7cc16ac9 Add warning about using sync-bn with zero initialized BN layers. Fixes #54
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 e42503e1d0 Update sotabench
5 years ago
Ross Wightman 2f41905ba5 Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.
5 years ago
Ross Wightman cc0b1f4130 Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
5 years ago
Ross Wightman 2a88412413
Update README.md
5 years ago
Ross Wightman d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
5 years ago
Ross Wightman 3eb4a96eda Update AugMix, JSD, etc comments and references
5 years ago
Ross Wightman a28117ea46 Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d
5 years ago
Ross Wightman cfa951bceb
Update README.md
5 years ago
Ross Wightman 833066b540 A few minor things in SplitBN
5 years ago
Ross Wightman 7547119891 Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
5 years ago
Ross Wightman 2e955cfd0c Update RandomErasing with some improved arg names, tweak to aspect range
5 years ago
Ross Wightman 3cc0f91e23 Fix augmix variable name scope overlap, default non-blended mode
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 8662454bd7 Plural for averaging script.
5 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
Ross Wightman 4666cc9aed Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script
5 years ago
Ross Wightman 53001dd292 ResNet / Res2Net additions:
5 years ago
Ross Wightman f96b3e5e92 InceptionResNetV2 torchscript compatible
5 years ago
Ross Wightman 19d93fe454 Add selecsls60 weights
5 years ago
Ross Wightman a7fe891a13
Update README.md
5 years ago
Ross Wightman e728d70831
Merge pull request #66 from rwightman/selecsls_updates
5 years ago
Ross Wightman 84ca3d1f4d Add SelecSLS to sotabench list
5 years ago
Ross Wightman 0062c15fb0 Update checkpoint url with modelzoo compatible ones.
5 years ago
Ross Wightman b5315e66b5 Streamline SelecSLS model without breaking checkpoint compat. Move cfg handling out of model class. Update feature/pooling behaviour to match current.
5 years ago