Update model.py

zsdonghao · May 31, 2017 · 19f2760 · 19f2760
1 parent d701068
commit 19f2760
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model.py b/model.py
@@ -813,6 +813,7 @@ def rnn_embed(input_seqs, is_train=True, reuse=False, return_embed=False):
                      name = 'rnn/wordembed')
         network = DynamicRNNLayer(network,
                      cell_fn = LSTMCell,
+                     cell_init_args = {'state_is_tuple' : True, 'reuse': reuse},
                      n_hidden = rnn_hidden_size,
                      dropout = (keep_prob if is_train else None),
                      initializer = w_init,