diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py index 2bff21cb78..832a56654e 100644 --- a/pytorch_lightning/callbacks/model_checkpoint.py +++ b/pytorch_lightning/callbacks/model_checkpoint.py @@ -1,3 +1,10 @@ +""" +Model Checkpointing +=================== + +Automatically save model checkpoints during training. +""" + import os import shutil import logging as log