Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/lightning/pytorch/core/module.py
Original file line number Diff line number Diff line change
Expand Up @@ -760,6 +760,10 @@ def training_step(self, batch, batch_idx):
When ``accumulate_grad_batches`` > 1, the loss returned here will be automatically
normalized by ``accumulate_grad_batches`` internally.

Note:
When the :meth:`training_step` is called, the training/eval mode of the model and submodules is
whatever it was when provided to `trainer.fit`.

"""
rank_zero_warn("`training_step` must be implemented to be used with the Lightning Trainer")

Expand Down
Loading