Skip to content

Commit 29af389

Browse files
carmoccalantiga
authored andcommitted
Update trainer.fit docstring (#17981)
(cherry picked from commit 4532ef7)
1 parent e4869ad commit 29af389

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/lightning/pytorch/trainer/trainer.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -510,14 +510,12 @@ def fit(
510510
511511
val_dataloaders: An iterable or collection of iterables specifying validation samples.
512512
513-
ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of two special
514-
keywords ``"last"`` and ``"hpc"``. If there is no checkpoint file at the path, an exception is raised.
515-
If resuming from mid-epoch checkpoint, training will start from the beginning of the next epoch.
516-
517-
datamodule: An instance of :class:`~lightning.pytorch.core.datamodule.LightningDataModule`.
518513
datamodule: A :class:`~lightning.pytorch.core.datamodule.LightningDataModule` that defines
519514
the :class:`~lightning.pytorch.core.hooks.DataHooks.train_dataloader` hook.
520515
516+
ckpt_path: Path/URL of the checkpoint from which training is resumed. Could also be one of two special
517+
keywords ``"last"`` and ``"hpc"``. If there is no checkpoint file at the path, an exception is raised.
518+
521519
Raises:
522520
TypeError:
523521
If ``model`` is not :class:`~lightning.pytorch.core.module.LightningModule` for torch version less than

0 commit comments

Comments
 (0)