6 Commits (16185270989eb25f5f15e7017b008103351fc771)

Author	SHA1	Message	Date
Ross Wightman	f0f9eccda8	Add --fuser arg to train/validate/benchmark scripts to select jit fuser type	3 years ago
Ross Wightman	e5da481073	Small post-merge tweak for freeze/unfreeze, add to __init__ for utils	3 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.	4 years ago
Ross Wightman	532e3b417d	Reorg of utils into separate modules	4 years ago