Skip to content

Commit 86c2b71

Browse files
authored
add training note
1 parent 37f0f30 commit 86c2b71

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/lightning/pytorch/core/module.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -760,6 +760,10 @@ def training_step(self, batch, batch_idx):
760760
When ``accumulate_grad_batches`` > 1, the loss returned here will be automatically
761761
normalized by ``accumulate_grad_batches`` internally.
762762
763+
Note:
764+
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`.
766+
763767
"""
764768
rank_zero_warn("`training_step` must be implemented to be used with the Lightning Trainer")
765769

0 commit comments

Comments
 (0)