From e0e98e05831862c5731e78bb32232c9ab2fbdfd9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Fri, 11 Jun 2021 21:06:22 +0200 Subject: [PATCH] update docs --- pytorch_lightning/trainer/connectors/checkpoint_connector.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/connectors/checkpoint_connector.py b/pytorch_lightning/trainer/connectors/checkpoint_connector.py index 8dea5e5af615a8..0cbc4386aeaab1 100644 --- a/pytorch_lightning/trainer/connectors/checkpoint_connector.py +++ b/pytorch_lightning/trainer/connectors/checkpoint_connector.py @@ -146,8 +146,9 @@ def restore_datamodule(self) -> None: def restore_model(self) -> None: """ - Restores a model's state from a PyTorch Lightning checkpoint. Hooks are called first, the weights get - restored last. + Restores a model's weights from a PyTorch Lightning checkpoint. Hooks are called first go give + the LightningModule a chance to modify the contents, then finally the model gets updated with + the loaded weights. """ if not self._loaded_checkpoint: return