File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
src/lightning/pytorch/trainer Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -526,7 +526,7 @@ def fit(
526
526
datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines
527
527
the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook.
528
528
529
- ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of two special
529
+ ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of three special
530
530
keywords ``"last"``, ``"hpc"`` and ``"registry"``.
531
531
Otherwise, if there is no checkpoint file at the path, an exception is raised.
532
532
@@ -535,12 +535,11 @@ def fit(
535
535
- registry: the model will be downloaded from the Lightning Model Registry with following notations:
536
536
537
537
- ``'registry'``: uses the latest/default version of default model set
538
- with ``Tainer (..., model_registry="my-model")``
538
+ with ``Trainer (..., model_registry="my-model")``
539
539
- ``'registry:model-name'``: uses the latest/default version of this model `model-name`
540
540
- ``'registry:model-name:version:v2'``: uses the specific version 'v2' of the model `model-name`
541
541
- ``'registry:version:v2'``: uses the default model set
542
- with ``Tainer(..., model_registry="my-model")`` and version 'v2'
543
-
542
+ with ``Trainer(..., model_registry="my-model")`` and version 'v2'
544
543
545
544
Raises:
546
545
TypeError:
You can’t perform that action at this time.
0 commit comments