Skip to content

Commit

Permalink
Fix dropout for recurrent dropout (keras-team#19397)
Browse files Browse the repository at this point in the history
  • Loading branch information
sachinprasadhs authored Mar 28, 2024
1 parent a994dfe commit c24b98a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keras/layers/rnn/dropout_rnn_cell.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit c24b98a

Please sign in to comment.