diff --git a/src/lightning/pytorch/trainer/trainer.py b/src/lightning/pytorch/trainer/trainer.py index 732521b1b0ce7..21c75c3096125 100644 --- a/src/lightning/pytorch/trainer/trainer.py +++ b/src/lightning/pytorch/trainer/trainer.py @@ -532,7 +532,7 @@ def fit( datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook. - ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of two special + ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of three special keywords ``"last"``, ``"hpc"`` and ``"registry"``. Otherwise, if there is no checkpoint file at the path, an exception is raised. @@ -541,12 +541,11 @@ def fit( - registry: the model will be downloaded from the Lightning Model Registry with following notations: - ``'registry'``: uses the latest/default version of default model set - with ``Tainer(..., model_registry="my-model")`` + with ``Trainer(..., model_registry="my-model")`` - ``'registry:model-name'``: uses the latest/default version of this model `model-name` - ``'registry:model-name:version:v2'``: uses the specific version 'v2' of the model `model-name` - ``'registry:version:v2'``: uses the default model set - with ``Tainer(..., model_registry="my-model")`` and version 'v2' - + with ``Trainer(..., model_registry="my-model")`` and version 'v2' Raises: TypeError: