Update pytorch_lightning/callbacks/base.py

carmocca · web-flow · commit 87b83a8ab50e · 2021-02-22T13:31:22.000+01:00
diff --git a/pytorch_lightning/callbacks/base.py b/pytorch_lightning/callbacks/base.py
@@ -193,6 +193,7 @@ def on_save_checkpoint(self, trainer, pl_module: LightningModule, checkpoint: Di
 
     def on_load_checkpoint(self, callback_state: Dict[str, Any]) -> None:
         """Called when loading a model checkpoint, use to reload state.
+
         Args:
             callback_state: the callback state returned by ``on_save_checkpoint``.
         """
