Skip to content

Commit

Permalink
release for TF1.2
Browse files Browse the repository at this point in the history
  • Loading branch information
zsdonghao committed Jun 18, 2017
1 parent 3c0ab0a commit ac7b989
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model.py
Original file line number Diff line number Diff line change
Expand Up @@ -813,7 +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},
cell_init_args = {'state_is_tuple' : True, 'reuse': reuse}, # for TF1.1, TF1.2 dont need to set reuse
n_hidden = rnn_hidden_size,
dropout = (keep_prob if is_train else None),
initializer = w_init,
Expand Down

0 comments on commit ac7b989

Please sign in to comment.