Quellcode durchsuchen

Merge pull request #1131 from tongda/master

add 'reuse' parameter to RNNCells to make code compatible with tensorflow master code
Neal Wu vor 8 Jahren
Ursprung
Commit
c774cc95e5
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      tutorials/rnn/ptb/ptb_word_lm.py

+ 1 - 1
tutorials/rnn/ptb/ptb_word_lm.py

@@ -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():