Set forget_bias=0 for static RNN implementation

Fixes #2612
This commit is contained in:
Alexandre Lissy 2019-12-20 13:28:42 +01:00
Родитель 551b3dd5f5
Коммит 5f003cfbd6
1 изменённых файлов: 1 добавлений и 0 удалений

Просмотреть файл

@ -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')