pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
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

Author

SHA1

Message

Date

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

1 Commits (4bb5e9b2247697ee5395975de1a07c058fb82061)