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.		2020-11-29 16:22:19 -08:00
..
__init__.py	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.	2020-11-29 16:22:19 -08:00
checkpoint_saver.py	…
cuda.py	Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.	2020-10-09 17:24:43 -07:00
distributed.py	…
jit.py	…
log.py	…
metrics.py	Fix topn metric view regression on PyTorch 1.7	2020-10-29 14:04:15 -07:00
misc.py	…
model.py	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.	2020-11-29 16:22:19 -08:00
model_ema.py	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.	2020-11-29 16:22:19 -08:00
summary.py	…