Skip to content

Commit 90e3679

Browse files
matsumotosanlantiga
authored andcommitted
Couple of Trainer.fit typos (#21080)
(cherry picked from commit 8d88f31)
1 parent 9d05e7a commit 90e3679

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/lightning/pytorch/trainer/trainer.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -526,7 +526,7 @@ def fit(
526526
datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines
527527
the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook.
528528
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
530530
keywords ``"last"``, ``"hpc"`` and ``"registry"``.
531531
Otherwise, if there is no checkpoint file at the path, an exception is raised.
532532
@@ -535,12 +535,11 @@ def fit(
535535
- registry: the model will be downloaded from the Lightning Model Registry with following notations:
536536
537537
- ``'registry'``: uses the latest/default version of default model set
538-
with ``Tainer(..., model_registry="my-model")``
538+
with ``Trainer(..., model_registry="my-model")``
539539
- ``'registry:model-name'``: uses the latest/default version of this model `model-name`
540540
- ``'registry:model-name:version:v2'``: uses the specific version 'v2' of the model `model-name`
541541
- ``'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'
544543
545544
Raises:
546545
TypeError:

0 commit comments

Comments
 (0)