Ross Wightman
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
Ross Wightman
31453b039e
Update Auto/RandAugment comments, README, more.
...
* Add a weighted choice option for RandAugment
* Adjust magnitude noise/std naming, config
2019-11-22 13:24:52 -08:00
Ross Wightman
4243f076f1
Adding RandAugment to AutoAugment impl, some tweaks to AA included
2019-11-21 21:14:33 -08:00
Ross Wightman
4002c0d4ce
Fix AutoAugment abs translate calc
2019-09-01 22:07:45 -07:00
Ross Wightman
b750b76f67
More AutoAugment work. Ready to roll...
2019-09-01 16:55:42 -07:00
Ross Wightman
25d2088d9e
Working on auto-augment
2019-08-31 23:09:48 -07:00
Ross Wightman
66634d2200
Add support to split random erasing blocks into randomly selected number with --recount arg. Fix random selection of aspect ratios.
2019-08-12 16:01:58 -07:00
Ross Wightman
65a634626f
Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print
2019-06-29 10:03:13 -07:00
Ross Wightman
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
2019-06-28 13:49:20 -07:00
Ross Wightman
aa4354f466
Big re-org, working towards making pip/module as 'timm'
2019-06-19 17:20:51 -07:00
