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.
2020-02-01 18:07:32 -08:00
91534522f9
Add newly added TF ported EfficientNet-B8 weights (RandAugment)
2020-02-01 18:01:14 -08:00
82dd60b33c
Rename results files for more clarity
2020-01-31 23:00:02 -08:00
5b7cc16ac9
Add warning about using sync-bn with zero initialized BN layers. Fixes #54
2020-01-31 11:44:24 -08:00
b18c19901e
Update README.md
...
Typo
2020-01-31 10:59:31 -08:00
12dbc74742
New ResNet50 JSD + RandAugment weights
2020-01-31 10:55:54 -08:00
e39aae56b4
Update README.md
...
Fix typo
2020-01-13 17:49:49 -08:00
7a17ee9c3b
Update README.md
2020-01-12 18:16:36 -08:00
e42503e1d0
Update sotabench
2020-01-12 18:00:12 -08:00
2f41905ba5
Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.
2020-01-12 17:55:58 -08:00
cc0b1f4130
Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
2020-01-12 17:52:19 -08:00
2a88412413
Update README.md
2020-01-11 12:08:15 -08:00
d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
...
AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more
2020-01-11 12:04:29 -08:00
3eb4a96eda
Update AugMix, JSD, etc comments and references
2020-01-11 12:02:05 -08:00
a28117ea46
Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d
2020-01-11 11:29:01 -08:00
cfa951bceb
Update README.md
2020-01-07 08:52:12 -08:00
833066b540
A few minor things in SplitBN
2020-01-05 20:07:03 -08:00
7547119891
Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
2020-01-05 19:58:59 -08:00
2e955cfd0c
Update RandomErasing with some improved arg names, tweak to aspect range
2020-01-05 14:31:48 -08:00
3cc0f91e23
Fix augmix variable name scope overlap, default non-blended mode
2020-01-05 14:27:27 -08:00
7622015258
Update README.md
2020-01-04 11:51:38 -08:00
ec0dd4053a
Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
2020-01-03 17:18:46 -08:00
8662454bd7
Plural for averaging script.
2020-01-03 15:00:20 -08:00
40fea63ebe
Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
2020-01-03 14:57:46 -08:00
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
2020-01-02 16:22:06 -08:00
53001dd292
ResNet / Res2Net additions:
...
* ResNet torchscript compat
* output_stride arg supported to limit network stride via dilations (support for dilation added to Res2Net)
* allow activation layer to be changed via act_layer arg
2020-01-01 17:15:56 -08:00
f96b3e5e92
InceptionResNetV2 torchscript compatible
2020-01-01 17:13:37 -08:00
19d93fe454
Add selecsls60 weights
2019-12-31 16:49:04 -08:00
a7fe891a13
Update README.md
...
Update readme with SelecSLS details.
2019-12-30 16:32:49 -08:00
e728d70831
Merge pull request #66 from rwightman/selecsls_updates
...
SelecSLS updates
2019-12-30 16:07:20 -08:00
84ca3d1f4d
Add SelecSLS to sotabench list
2019-12-30 16:04:39 -08:00
0062c15fb0
Update checkpoint url with modelzoo compatible ones.
2019-12-30 15:59:19 -08:00
b5315e66b5
Streamline SelecSLS model without breaking checkpoint compat. Move cfg handling out of model class. Update feature/pooling behaviour to match current.
2019-12-30 15:44:47 -08:00
d59a756c16
Run PyCharm autoformat on selecsls and change mix cap variables and model names to all lower
2019-12-30 14:30:46 -08:00
fb3a0f4bb8
Merge pull request #65 from mehtadushy/selecsls
...
Incorporate SelecSLS Models
2019-12-30 14:23:53 -08:00
0554b7942c
Update README.md
2019-12-29 09:15:00 -08:00
a4497afa21
Update sotabench.py
2019-12-28 23:06:27 -08:00
53f578e16f
Update README with B3 training details
2019-12-28 17:51:08 -08:00
19fc205a4d
Update comments on the new SE-ResNeXt26 models
2019-12-28 17:33:10 -08:00
8894a6bc08
Update sotabench.py with latest seresnext26 and efficientnet_b3 models
2019-12-28 17:25:53 -08:00
acc3ed2b8c
Add EfficientNet-B3 weights, trained from scratch with RA.
2019-12-28 17:24:15 -08:00
2404361f62
correct asset paths
2019-12-28 23:32:20 +01:00
31939311f6
Added SelecSLS models
2019-12-28 23:06:00 +01:00
1f4498f217
Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d
2019-12-28 11:43:50 -08:00
32012a44fd
Added SelecSLS model
2019-12-28 20:41:55 +01:00
73b78459dc
Add update RandAugment MixNet-XL weights
2019-12-24 10:08:24 -08:00
3afc2a4dc0
Some cleanup/improvements to AugMix impl:
...
* make 'increasing' levels for Contrast, Color, Brightness, Saturation ops
* remove recursion from faster blending mix
* add config striing parsing for AugMix
2019-12-20 23:04:11 -08:00
232ab7fb12
Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl
2019-12-20 23:04:11 -08:00
ff8688ca3d
Merge pull request #62 from rwightman/reduce-bn
...
Distribute BatchNorm stats
2019-12-19 23:22:16 -08:00
5719b493ad
Missed update dist-bn logic for EMA model
2019-12-19 23:03:04 -08:00
Ross Wightman
Ross Wightman
Dushyant Mehta