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