William Falcon cf182e80fc Finish Allow on_save_checkpoint... (#3688) ... * Finish #3562 * Apply suggestions from code review * Apply suggestions from code review * fix tests * Finish #3562 * Apply suggestions from code review * Apply suggestions from code review * fix tests * fix structure * fix structure * make save_last test pass * unnecessary global rank check * fix test * update test * update test * test * test * run save on all * remove assert * tracking saves * check if fails * test * clean up * adjust horovod test * clean up * remove unnecessary makdirs * change * undo * debug * debug * debug * debug * mock * undo debug code * add extra assertions * test Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Adrian Wälchli <adrian.waelchli@inf.unibe.ch> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2020-09-30 16:15:29 -04:00
..
__init__.py	Refactor GPUStatsMonitor to improve training speed (#3257)	2020-09-04 06:02:16 -04:00
base.py	callback method for on_save_checkpoint (#2501)	2020-08-28 16:50:52 +02:00
early_stopping.py	ref: result 1/n (make monitor default to checkpoint_on to simplify re… (#3571)	2020-09-20 22:58:43 -04:00
gpu_stats_monitor.py	Split GPUStatsMonitor function (#3644)	2020-09-25 07:30:30 +02:00
gpu_usage_logger.py	Refactor GPUStatsMonitor to improve training speed (#3257)	2020-09-04 06:02:16 -04:00
gradient_accumulation_scheduler.py	Add stronger typing to gradient accumulation scheduler callback (#3558)	2020-09-23 20:22:10 +02:00
lr_logger.py	Changed LearningRateLogger to LearningRateMonitor (#3251)	2020-09-03 18:17:15 +00:00
lr_monitor.py	Changed LearningRateLogger to LearningRateMonitor (#3251)	2020-09-03 18:17:15 +00:00
model_checkpoint.py	Finish Allow on_save_checkpoint... (#3688)	2020-09-30 16:15:29 -04:00
progress.py	docs: use ref for anchor links, fix a few typo (#3486)	2020-09-13 21:04:21 -04:00