8 Commits (53e66d223f0072558b2bf8426e208c01e4398f31)

Author	SHA1	Message	Date
Ross Wightman	87939e6fab	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	3 years ago
Ross Wightman	0dbd9352ce	Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry...	3 years ago
Ross Wightman	f0f9eccda8	Add --fuser arg to train/validate/benchmark scripts to select jit fuser type	4 years ago
Ross Wightman	e5da481073	Small post-merge tweak for freeze/unfreeze, add to __init__ for utils	4 years ago
Ross Wightman	7c97e66f7c	Remove commented code, add more consistent seed fn	4 years ago
Ross Wightman	4f49b94311	Initial AGC impl. Still testing.	4 years ago
Ross Wightman	27bbc70d71	Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.	5 years ago
Ross Wightman	532e3b417d	Reorg of utils into separate modules	5 years ago