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