"git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "b985609bc1d8bd4bd5f02619f67c4533ddc88421"
Commit f6e23e56 authored by Neal Wu's avatar Neal Wu
Browse files

Revert "add 'reuse' parameter to RNNCells to make code compatible with tensorflow master code"

parent 2c6d74b7
...@@ -110,7 +110,7 @@ class PTBModel(object): ...@@ -110,7 +110,7 @@ class PTBModel(object):
# different than reported in the paper. # different than reported in the paper.
def lstm_cell(): def lstm_cell():
return tf.contrib.rnn.BasicLSTMCell( return tf.contrib.rnn.BasicLSTMCell(
size, forget_bias=0.0, state_is_tuple=True, reuse=tf.get_variable_scope().reuse) size, forget_bias=0.0, state_is_tuple=True)
attn_cell = lstm_cell attn_cell = lstm_cell
if is_training and config.keep_prob < 1: if is_training and config.keep_prob < 1:
def attn_cell(): def attn_cell():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment