Ross Wightman 4d2056722a Mixup and prefetcher improvements ... * Do mixup in custom collate fn if prefetcher enabled, reduces performance impact * Move mixup code to own file * Add arg to disable prefetcher * Fix no cuda transfer when prefetcher off * Random erasing when prefetcher off wasn't changed to match new args, fixed * Default random erasing to off (prob = 0.) for train		2019-05-18 22:17:51 -07:00
..
__init__.py	Mixup and prefetcher improvements	2019-05-18 22:17:51 -07:00
config.py	Some transform/data/loader refactoring, hopefully didn't break things	2019-05-16 22:52:17 -07:00
constants.py	Some transform/data/loader refactoring, hopefully didn't break things	2019-05-16 22:52:17 -07:00
dataset.py	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	2019-03-11 22:17:42 -07:00
distributed_sampler.py	Add distributed sampler that maintains order of original dataset (for validation)	2019-04-22 17:44:53 -07:00
loader.py	Mixup and prefetcher improvements	2019-05-18 22:17:51 -07:00
mixup.py	Mixup and prefetcher improvements	2019-05-18 22:17:51 -07:00
random_erasing.py	Change args for RandomErasing so only one required for pixel/color mode	2019-05-18 12:29:30 -07:00
transforms.py	Mixup and prefetcher improvements	2019-05-18 22:17:51 -07:00