Merge pull request #2613 from lissyx/fix-tflite-forget_bias

Set forget_bias=0 for static RNN implementation
2019-12-20 14:54:54 +01:00 · 2019-12-20 14:54:54 +01:00 · 1d0035ce7f
commit 1d0035ce7f
parent 551b3dd5f5 5f003cfbd6
1 changed files with 1 additions and 0 deletions
--- a/DeepSpeech.py
+++ b/DeepSpeech.py
@ -126,6 +126,7 @@ def rnn_impl_static_rnn(x, seq_length, previous_state, reuse):
    with tfv1.variable_scope('cudnn_lstm/rnn/multi_rnn_cell'):
        # Forward direction cell:
        fw_cell = tfv1.nn.rnn_cell.LSTMCell(Config.n_cell_dim,
+                                            forget_bias=0,
                                            reuse=reuse,
                                            name='cudnn_compatible_lstm_cell')