11 Commits (multi-weight)

Author	SHA1	Message	Date
Ross Wightman	87939e6fab	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	3 years ago
Ross Wightman	656757d26b	Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.	4 years ago
Ross Wightman	ed41d32637	Add repr to auto_augment and random_erasing impl	4 years ago
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	5 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	0161de0127	Switch RandoErasing back to on GPU normal sampling	6 years ago
Ross Wightman	66634d2200	Add support to split random erasing blocks into randomly selected number with --recount arg. Fix random selection of aspect ratios.	6 years ago
Ross Wightman	6946281fde	Experimenting with random erasing changes	6 years ago
Ross Wightman	65a634626f	Switch random erasing to doing normal_() on CPU to avoid instability, remove a debug print	6 years ago
Ross Wightman	aa4354f466	Big re-org, working towards making pip/module as 'timm'	6 years ago