experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source 
Set forget_bias=0 for static RNN implementation
This commit is contained in:
lissyx 2019-12-20 14:54:54 +01:00 committed by GitHub
commit 1d0035ce7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
DeepSpeech.py
View File

@ -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'): with tfv1.variable_scope('cudnn_lstm/rnn/multi_rnn_cell'):
# Forward direction cell: # Forward direction cell:
fw_cell = tfv1.nn.rnn_cell.LSTMCell(Config.n_cell_dim, fw_cell = tfv1.nn.rnn_cell.LSTMCell(Config.n_cell_dim,
forget_bias=0,
reuse=reuse, reuse=reuse,
name='cudnn_compatible_lstm_cell') name='cudnn_compatible_lstm_cell')