From c24b98a3528c5346370b6c7158674476ebf1c19e Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Thu, 28 Mar 2024 13:09:18 -0700 Subject: [PATCH] Fix dropout for recurrent dropout (#19397) --- keras/layers/rnn/dropout_rnn_cell.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras/layers/rnn/dropout_rnn_cell.py b/keras/layers/rnn/dropout_rnn_cell.py index 5b0b83ac8c9..a4d4d69fa55 100644 --- a/keras/layers/rnn/dropout_rnn_cell.py +++ b/keras/layers/rnn/dropout_rnn_cell.py @@ -35,7 +35,7 @@ def get_recurrent_dropout_mask(self, step_input): if self._recurrent_dropout_mask is None and self.recurrent_dropout > 0: ones = ops.ones_like(step_input) self._recurrent_dropout_mask = backend.random.dropout( - ones, rate=self.dropout, seed=self.seed_generator + ones, rate=self.recurrent_dropout, seed=self.seed_generator ) return self._recurrent_dropout_mask