7 Commits (19816fe2262a62c172458e36a53ed65376b65318)

Author	SHA1	Message	Date
Ross Wightman	4ca52d73d8	Add separate set and update method to ModelEmaV2	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	9214ca0716	Simplifying EMA...	4 years ago
Ross Wightman	4a3df7842a	Fix topn metric view regression on PyTorch 1.7	4 years ago
Ross Wightman	80078c47bb	Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.	4 years ago
Ross Wightman	fcb6258877	Add missing leaky_relu layer factory defn, update Apex/Native loss scaler interfaces to support unscaled grad clipping. Bump ver to 0.2.2 for pending release.	4 years ago
Ross Wightman	532e3b417d	Reorg of utils into separate modules	4 years ago