From 1da1898d41d960cb8f64d4cdde834f8606c369cc Mon Sep 17 00:00:00 2001 From: Guillaume Tauzin Date: Sun, 6 Jun 2021 20:31:14 +0200 Subject: [PATCH] [docs] Fix truncated_bptt_steps docs (#7846) --- docs/source/common/lightning_module.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/common/lightning_module.rst b/docs/source/common/lightning_module.rst index ad8652f446..1689100511 100644 --- a/docs/source/common/lightning_module.rst +++ b/docs/source/common/lightning_module.rst @@ -1036,7 +1036,7 @@ recurrent network trajectories." def training_step(self, batch, batch_idx, hiddens): # the training step must be updated to accept a ``hiddens`` argument # hiddens are the hiddens from the previous truncated backprop step - out, hiddens = self.lstm(data, hiddens) + out, (hiddens, _) = self.lstm(data, hiddens) return { "loss": ..., "hiddens": hiddens