Skip to content

Commit 37bd23e

Browse files
authored
Update src/lightning/pytorch/core/module.py
1 parent 3302098 commit 37bd23e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/core/module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -762,7 +762,7 @@ def training_step(self, batch, batch_idx):
762762
763763
Note:
764764
When the :meth:`training_step` is called, the training/eval mode of the model and submodules is
765-
whatever it was when provided to `trainer.fit`.
765+
whatever it was when provided to :meth:`~lightning.pytorch.trainer.trainer.Trainer.fit`.
766766
767767
"""
768768
rank_zero_warn("`training_step` must be implemented to be used with the Lightning Trainer")

0 commit comments

Comments
 (0)