6 Commits

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	2019-05-18 22:17:51 -07:00
Ross Wightman	fee607edf6	Mixup implemention in progress ... * initial impl w/ label smoothing converging, but needs more testing	2019-05-13 19:05:40 -07:00
Ross Wightman	db1fe34d0c	Update a few comment, add some references	2019-04-12 23:16:49 -07:00
Ross Wightman	f1cd1a5ce3	Cleanup CheckpointSaver, add support for increasing or decreasing metric, switch to prec1 metric in train loop	2019-04-07 10:22:55 -07:00
Ross Wightman	a336e5bff3	Minor updates	2019-02-08 20:56:24 -08:00
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	2019-02-01 22:07:34 -08:00