[docs] Fix truncated_bptt_steps docs (#7846)

2021-06-06 20:31:14 +02:00 · 2021-06-06 20:31:14 +02:00 · 1da1898d41
parent 10839376e2
commit 1da1898d41
1 changed files with 1 additions and 1 deletions
--- 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