Skip to content

Instantly share code, notes, and snippets.

@3h4

3h4/6-1-add-dropout-multi-lstm.py

Created November 19, 2016 11:51
Show Gist options
  • Save 3h4/341cf83ac05e8ef66216f75d906b4839 to your computer and use it in GitHub Desktop.
Save 3h4/341cf83ac05e8ef66216f75d906b4839 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
6-1-add-dropout-multi-lstm.py
cell = tf.nn.rnn_cell.LSTMCell(state_size, state_is_tuple=True)
cell = tf.nn.rnn_cell.DropoutWrapper(cell, output_keep_prob=0.5)
cell = tf.nn.rnn_cell.MultiRNNCell([cell] * num_layers, state_is_tuple=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment