From b29ac5c8732bb34dc94ae7517771b0119ccf5894 Mon Sep 17 00:00:00 2001 From: Mokhov Andrey Date: Tue, 11 Apr 2023 08:01:25 +0300 Subject: [PATCH] lab04: states for LSTM --- labs/OATD_LR4_metod.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/labs/OATD_LR4_metod.ipynb b/labs/OATD_LR4_metod.ipynb index 7d7f1a0..128279a 100644 --- a/labs/OATD_LR4_metod.ipynb +++ b/labs/OATD_LR4_metod.ipynb @@ -918,7 +918,7 @@ " if states is None:\n", " states = self.gru.get_initial_state(x)\n", "\n", - " x, states = self.gru(x, initial_state=states, training=training)\n", + " x, *states = self.gru(x, initial_state=states, training=training)\n", " x = self.dense(x, training=training)\n", "\n", " if return_state:\n",