Commit c774cc95 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1131 from tongda/master

add 'reuse' parameter to RNNCells to make code compatible with tensorflow master code
parents 88111129 bae323af
......@@ -110,7 +110,7 @@ class PTBModel(object):
# different than reported in the paper.
def lstm_cell():
return tf.contrib.rnn.BasicLSTMCell(
size, forget_bias=0.0, state_is_tuple=True)
size, forget_bias=0.0, state_is_tuple=True, reuse=tf.get_variable_scope().reuse)
attn_cell = lstm_cell
if is_training and config.keep_prob < 1:
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