From e2437ebf3ffc873f1d7a1fa0d4c00c7f133246a6 Mon Sep 17 00:00:00 2001
From: Paul Baumstark <41694868+pbmstrk@users.noreply.github.com>
Date: Mon, 5 Oct 2020 04:34:27 +0100
Subject: [PATCH] Update lightning_module.rst (#3854)

`*_epoch_out` methods expects a return of None.
---
 docs/source/lightning_module.rst | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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
 ^^^^^^^^^^^^^^^^^^^^^^^^^^