Commit Graph

309 Commits (7011cd0902009baaa3ceb4754fe7f379361b8a6a)
 

Author SHA1 Message Date
Ross Wightman 7011cd0902 A little bit of ECA cleanup
5 years ago
Ross Wightman 46471df7b2 Merge pull request #82 from VRandme/eca
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
Chris Ha e6a762346a Implement Adaptive Kernel selection
5 years ago
Chris Ha 6db087a1ff Merge remote-tracking branch 'upstream/master' into eca
5 years ago
Chris Ha 904c618040 Update EcaModule.py
5 years ago
Chris Ha db91ba053b EcaModule(CamelCase)
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
Chris Ha d04ff95eda Merge branch 'master' into eca
5 years ago
Chris Ha d63ae121d5 Clean up eca_module code
5 years ago
Chris Ha b0d45fd09c Merge remote-tracking branch 'upstream/master'
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
Chris Ha f87fcd7e88 Implement Eca modules
5 years ago
Chris Ha 697e05cb3e Merge remote-tracking branch 'upstream/master'
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
Chris Ha 89a32454e1 Merge remote-tracking branch 'upstream/master'
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
Chris Ha 0697ab183b Update .gitignore
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