fa28067704
Add more augmentation arguments, including a no_aug disable flag. Fix #209
4 years ago
6c17d57a2c
Fix some attributions, add copyrights to some file docstrings
4 years ago
a7f570c9b7
added MultiEpochsDataLoader
5 years ago
d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
...
AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more
5 years ago
7547119891
Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
5 years ago
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
232ab7fb12
Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl
5 years ago
b750b76f67
More AutoAugment work. Ready to roll...
5 years ago
25d2088d9e
Working on auto-augment
5 years ago
f37e633e9b
Merge remote-tracking branch 'origin/re-exp' into opt
5 years ago
daeaa113e2
Add initial sotabench attempt. Split create_transform out of create_loader. Update requirements.txt
5 years ago
66634d2200
Add support to split random erasing blocks into randomly selected number with --recount arg. Fix random selection of aspect ratios.
5 years ago
dfa9298b4e
Add MixNet ( https://arxiv.org/abs/1907.09595 ) with pretrained weights converted from Tensorflow impl
...
* refactor 'same' convolution and add helper to use MixedConv2d when needed
* improve performance of 'same' padding for cases that can be handled statically
* add support for extra exp, pw, and dw kernel specs with grouping support to decoder/string defs for MixNet
* shuffle some args for a bit more consistency, a little less clutter overall in gen_efficientnet.py
5 years ago
6cdf35e670
Add explicit half/fp16 support to loader and validation script
6 years ago
c6b32cbe73
A number of tweaks to arguments, epoch handling, config
...
* reorganize train args
* allow resolve_data_config to be used with dict args, not just arparse
* stop incrementing epoch before save, more consistent naming vs csv, etc
* update resume and start epoch handling to match above
* stop auto-incrementing epoch in scheduler
6 years ago
aa4354f466
Big re-org, working towards making pip/module as 'timm'
6 years ago
Ross Wightman