lightning/tests/callbacks
Jeremy Jordan 1e1a42260a
add option to log momentum (#4384)
* add option to log momentum

* add docstring

* refactor for cleanliness

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>
2020-10-28 21:56:58 +05:30
..
__init__.py Progress bar callback (#1450) 2020-04-23 20:46:18 -04:00
test_callback_hook_outputs.py notices (#4118) 2020-10-13 07:18:07 -04:00
test_callbacks.py Add optimizer hooks in callbacks (#4379) 2020-10-28 13:15:22 +01:00
test_early_stopping.py Add dirpath and filename parameter in ModelCheckpoint (#4213) 2020-10-23 09:59:12 +05:30
test_gpu_stats_monitor.py limit monitor callback with log_every_n_steps (#3881) 2020-10-22 16:38:03 +05:30
test_lr_monitor.py add option to log momentum (#4384) 2020-10-28 21:56:58 +05:30
test_progress_bar.py Add dirpath and filename parameter in ModelCheckpoint (#4213) 2020-10-23 09:59:12 +05:30