From 5a287a65e5a6db88f3b20de840edb23f92170499 Mon Sep 17 00:00:00 2001 From: Reuben Morais Date: Mon, 7 Oct 2019 21:06:29 +0200 Subject: [PATCH] Set forget_bias=0 in CPU graph for compatibility with CudnnRNN --- DeepSpeech.py | 1 + 1 file changed, 1 insertion(+) diff --git a/DeepSpeech.py b/DeepSpeech.py index a2dd045a..cea43e2a 100755 --- a/DeepSpeech.py +++ b/DeepSpeech.py @@ -82,6 +82,7 @@ def dense(name, x, units, dropout_rate=None, relu=True): def rnn_impl_lstmblockfusedcell(x, seq_length, previous_state, reuse): with tfv1.variable_scope('cudnn_lstm/rnn/multi_rnn_cell/cell_0'): fw_cell = tf.contrib.rnn.LSTMBlockFusedCell(Config.n_cell_dim, + forget_bias=0, reuse=reuse, name='cudnn_compatible_lstm_cell')