diff --git a/docs/source/lightning_module.rst b/docs/source/lightning_module.rst index a4e9dbe183..dfa8977299 100644 --- a/docs/source/lightning_module.rst +++ b/docs/source/lightning_module.rst @@ -448,7 +448,7 @@ The matching pseudocode is: optimizer.step() optimizer.zero_grad() - epoch_out = training_epoch_end(outs) + training_epoch_end(outs) Training with DataParallel ^^^^^^^^^^^^^^^^^^^^^^^^^^