Chris Ha
|
9e3987aefd
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Ross Wightman
|
f098fda2ca
|
Add map_location='cpu' to ModelEma resume, should improve #72
|
6 years ago |
|
Ross Wightman
|
b9496995d8
|
Add L2-475 PyTorch preprocessing result, update sotabench for new weights
|
6 years ago |
|
Ross Wightman
|
ba15ca47e8
|
Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
|
6 years ago |
|
Chris Ha
|
d5a577bb84
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Chris Ha
|
04a29656dc
|
Update .gitignore
add testing notebook into gitignore
|
6 years ago |
|
Ross Wightman
|
d0eb59ef46
|
Remove unused default_init for EfficientNets, experimenting with fanout calc for #84
|
6 years ago |
|
Ross Wightman
|
cade829105
|
Add EfficientNet-ES to sotabench
|
6 years ago |
|
Chris Ha
|
99142b378e
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Ross Wightman
|
5eb0e363a6
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
5c4991a088
|
Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
|
6 years ago |
|
Chris Ha
|
b0d45fd09c
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Ross Wightman
|
d66819d1f3
|
Indentation mistake. Fixes #81
|
6 years ago |
|
Ross Wightman
|
65cda1c63f
|
Merge pull request #83 from andravin/validation-batch-size-multiplier
Added commandline argument validation-batch-size-multiplier.
|
6 years ago |
|
Chris Ha
|
697e05cb3e
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Andrew Lavin
|
b72013def8
|
Added commandline argument validation-batch-size-multiplier with default set to 1.
|
6 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)
|
6 years ago |
|
Ross Wightman
|
5c85389e26
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
820b73d280
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
82c0a2ff12
|
Update README.md
Fix relative paths (I think)
|
6 years ago |
|
Ross Wightman
|
1ffd2d0694
|
Add results/README.md
|
6 years ago |
|
Chris Ha
|
89a32454e1
|
Merge remote-tracking branch 'upstream/master'
|
6 years ago |
|
Ross Wightman
|
fd98fb33c5
|
Update sotabench with tf_efficientnet_b8 model
|
6 years ago |
|
Ross Wightman
|
7c88356682
|
Add update results on ImageNet validation, ImageNetV2, ImageNet-A, and ImageNet-Sketch for all models
|
6 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.
|
6 years ago |
|
Ross Wightman
|
91534522f9
|
Add newly added TF ported EfficientNet-B8 weights (RandAugment)
|
6 years ago |
|
Ross Wightman
|
82dd60b33c
|
Rename results files for more clarity
|
6 years ago |
|
Ross Wightman
|
5b7cc16ac9
|
Add warning about using sync-bn with zero initialized BN layers. Fixes #54
|
6 years ago |
|
Ross Wightman
|
b18c19901e
|
Update README.md
Typo
|
6 years ago |
|
Ross Wightman
|
12dbc74742
|
New ResNet50 JSD + RandAugment weights
|
6 years ago |
|
Chris Ha
|
0697ab183b
|
Update .gitignore
|
6 years ago |
|
Ross Wightman
|
e39aae56b4
|
Update README.md
Fix typo
|
6 years ago |
|
Ross Wightman
|
7a17ee9c3b
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
e42503e1d0
|
Update sotabench
|
6 years ago |
|
Ross Wightman
|
2f41905ba5
|
Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.
|
6 years ago |
|
Ross Wightman
|
cc0b1f4130
|
Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
|
6 years ago |
|
Ross Wightman
|
2a88412413
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
d9a6a9d0af
|
Merge pull request #74 from rwightman/augmix-jsd
AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more
|
6 years ago |
|
Ross Wightman
|
3eb4a96eda
|
Update AugMix, JSD, etc comments and references
|
6 years ago |
|
Ross Wightman
|
a28117ea46
|
Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d
|
6 years ago |
|
Ross Wightman
|
cfa951bceb
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
833066b540
|
A few minor things in SplitBN
|
6 years ago |
|
Ross Wightman
|
7547119891
|
Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
|
6 years ago |
|
Ross Wightman
|
2e955cfd0c
|
Update RandomErasing with some improved arg names, tweak to aspect range
|
6 years ago |
|
Ross Wightman
|
3cc0f91e23
|
Fix augmix variable name scope overlap, default non-blended mode
|
6 years ago |
|
Ross Wightman
|
7622015258
|
Update README.md
|
6 years ago |
|
Ross Wightman
|
ec0dd4053a
|
Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
|
6 years ago |
|
Ross Wightman
|
8662454bd7
|
Plural for averaging script.
|
6 years ago |
|
Ross Wightman
|
40fea63ebe
|
Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
|
6 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
|
6 years ago |
