Skip to content

Instantly share code, notes, and snippets.

@ysyun
Created August 28, 2018 06:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ysyun/79c2ee02ba2ab8f0a55aa4a2234f1958 to your computer and use it in GitHub Desktop.
Save ysyun/79c2ee02ba2ab8f0a55aa4a2234f1958 to your computer and use it in GitHub Desktop.
large train yml of data2vis
model: AttentionSeq2Seq
model_params:
attention.class: seq2seq.decoders.attention.AttentionLayerBahdanau
attention.params:
num_units: 512
bridge.class: seq2seq.models.bridges.ZeroBridge
embedding.dim: 512
encoder.class: seq2seq.encoders.BidirectionalRNNEncoder
encoder.params:
rnn_cell:
cell_class: LSTMCell
cell_params:
num_units: 512
dropout_input_keep_prob: 0.5
dropout_output_keep_prob: 1.0
num_layers: 2
decoder.class: seq2seq.decoders.AttentionDecoder
decoder.params:
rnn_cell:
cell_class: LSTMCell
cell_params:
num_units: 512
dropout_input_keep_prob: 0.5
dropout_output_keep_prob: 1.0
num_layers: 4
optimizer.name: Adam
optimizer.params:
epsilon: 0.0000008
optimizer.learning_rate: 0.0001
source.max_seq_len: 500
source.reverse: false
target.max_seq_len: 500
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment