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 @@ -532,7 +532,7 @@ def fit(
532
532
datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines
533
533
the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook.
534
534
535
- ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of two special
535
+ ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of three special
536
536
keywords ``"last"``, ``"hpc"`` and ``"registry"``.
537
537
Otherwise, if there is no checkpoint file at the path, an exception is raised.
538
538
@@ -541,12 +541,11 @@ def fit(
541
541
- registry: the model will be downloaded from the Lightning Model Registry with following notations:
542
542
543
543
- ``'registry'``: uses the latest/default version of default model set
544
- with ``Tainer (..., model_registry="my-model")``
544
+ with ``Trainer (..., model_registry="my-model")``
545
545
- ``'registry:model-name'``: uses the latest/default version of this model `model-name`
546
546
- ``'registry:model-name:version:v2'``: uses the specific version 'v2' of the model `model-name`
547
547
- ``'registry:version:v2'``: uses the default model set
548
- with ``Tainer(..., model_registry="my-model")`` and version 'v2'
549
-
548
+ with ``Trainer(..., model_registry="my-model")`` and version 'v2'
550
549
551
550
Raises:
552
551
TypeError:
You can’t perform that action at this time.
0 commit comments