1e1a42260a
* add option to log momentum * add docstring * refactor for cleanliness Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
base.py | ||
early_stopping.py | ||
gpu_stats_monitor.py | ||
gradient_accumulation_scheduler.py | ||
lr_monitor.py | ||
model_checkpoint.py | ||
progress.py |