зеркало из https://github.com/mozilla/DeepSpeech.git
Set forget_bias=0 in CPU graph for compatibility with CudnnRNN
This commit is contained in:
Родитель
fb611efd00
Коммит
5a287a65e5
|
@ -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')
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче