You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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		6 years ago
..
__init__.py	Mixup and prefetcher improvements	6 years ago
config.py	Some transform/data/loader refactoring, hopefully didn't break things	6 years ago
constants.py	Some transform/data/loader refactoring, hopefully didn't break things	6 years ago
dataset.py	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	6 years ago
distributed_sampler.py	Add distributed sampler that maintains order of original dataset (for validation)	6 years ago
loader.py	Mixup and prefetcher improvements	6 years ago
mixup.py	Mixup and prefetcher improvements	6 years ago
random_erasing.py	Change args for RandomErasing so only one required for pixel/color mode	6 years ago
transforms.py	Mixup and prefetcher improvements	6 years ago