From 0ac9e2ae608c173b45bed8ab947c2be3f4cd9b53 Mon Sep 17 00:00:00 2001 From: Caleb Robinson Date: Tue, 20 Jul 2021 01:09:31 -0700 Subject: [PATCH] Adding appends to some of the pseudocode blocks (#8427) --- docs/source/common/lightning_module.rst | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/source/common/lightning_module.rst b/docs/source/common/lightning_module.rst index 84ffb7cec8..fe66ffacb9 100644 --- a/docs/source/common/lightning_module.rst +++ b/docs/source/common/lightning_module.rst @@ -213,6 +213,7 @@ Here's the pseudocode of what it does under the hood: for batch in train_dataloader: # forward out = training_step(val_batch) + outs.append(out) # clear gradients optimizer.zero_grad() @@ -250,6 +251,7 @@ The matching pseudocode is: for batch in train_dataloader: # forward out = training_step(val_batch) + outs.append(out) # clear gradients optimizer.zero_grad()