Ross Wightman
|
87939e6fab
|
Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.
|
2022-09-23 16:08:59 -07:00 |
|
Ross Wightman
|
656757d26b
|
Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.
|
2022-01-14 16:28:27 -08:00 |
|
Ross Wightman
|
ed41d32637
|
Add repr to auto_augment and random_erasing impl
|
2021-10-28 17:33:36 -07:00 |
|
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
2020-07-27 13:44:56 -07:00 |
|
Ross Wightman
|
7547119891
|
Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
|
2020-01-05 19:58:59 -08:00 |
|
Ross Wightman
|
2e955cfd0c
|
Update RandomErasing with some improved arg names, tweak to aspect range
|
2020-01-05 14:31:48 -08:00 |
|
Ross Wightman
|
0161de0127
|
Switch RandoErasing back to on GPU normal sampling
|
2019-12-05 22:35:08 -08: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
|
6946281fde
|
Experimenting with random erasing changes
|
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
|
aa4354f466
|
Big re-org, working towards making pip/module as 'timm'
|
2019-06-19 17:20:51 -07:00 |
|