lightning/pytorch_lightning/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 notices (#4118) 2020-10-13 07:18:07 -04:00
base.py Add optimizer hooks in callbacks (#4379) 2020-10-28 13:15:22 +01:00
early_stopping.py Fix info message when EarlyStopping 'mode' not provided [ci skip] (#4282) 2020-10-21 23:44:13 +05:30
gpu_stats_monitor.py limit monitor callback with log_every_n_steps (#3881) 2020-10-22 16:38:03 +05:30
gradient_accumulation_scheduler.py Add stronger typing to gradient accumulation scheduler callback (#3558) 2020-09-23 20:22:10 +02:00
lr_monitor.py add option to log momentum (#4384) 2020-10-28 21:56:58 +05:30
model_checkpoint.py Add "monitor" to saved ModelCheckpoints (#4383) 2020-10-28 15:21:08 +05:30
progress.py outputs in __batch_end hooks (#3966) 2020-10-07 21:48:38 -04:00