Update lightning_module.rst (#3854)

`*_epoch_out` methods expects a return of None.
2020-10-05 04:34:27 +01:00 · 2020-10-05 04:34:27 +01:00 · e2437ebf3f
parent d787208e76
commit e2437ebf3f
1 changed files with 1 additions and 1 deletions
--- 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
 ^^^^^^^^^^^^^^^^^^^^^^^^^^