Skip to content

Commit 8d88f31

Browse files
authored
Couple of Trainer.fit typos (#21080)
1 parent 8d1a734 commit 8d88f31

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
@@ -532,7 +532,7 @@ def fit(
532532
datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines
533533
the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook.
534534
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
536536
keywords ``"last"``, ``"hpc"`` and ``"registry"``.
537537
Otherwise, if there is no checkpoint file at the path, an exception is raised.
538538
@@ -541,12 +541,11 @@ def fit(
541541
- registry: the model will be downloaded from the Lightning Model Registry with following notations:
542542
543543
- ``'registry'``: uses the latest/default version of default model set
544-
with ``Tainer(..., model_registry="my-model")``
544+
with ``Trainer(..., model_registry="my-model")``
545545
- ``'registry:model-name'``: uses the latest/default version of this model `model-name`
546546
- ``'registry:model-name:version:v2'``: uses the specific version 'v2' of the model `model-name`
547547
- ``'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'
550549
551550
Raises:
552551
TypeError:

0 commit comments

Comments
 (0)