Added a config file example showing how to use attention.

mckinziebrandon · May 17, 2017 · 77c7f8a · 77c7f8a
1 parent 52da99d
commit 77c7f8a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/configs/example_attention.yml b/configs/example_attention.yml
@@ -0,0 +1,17 @@
+model: DynamicBot
+dataset: Cornell
+model_params:
+  base_cell: LSTMCell
+  ckpt_dir: out/cornell
+  attention_mechanism: BahdanauAttention
+  decoder.class: AttentionDecoder
+  encoder.class: BasicEncoder
+  batch_size: 256
+  embed_size: 128
+  num_layers: 1
+  state_size: 512
+  steps_per_ckpt: 250
+dataset_params:
+  data_dir: /home/brandon/Datasets/cornell  # Change to your path!
+  vocab_size: 52000
+  max_seq_len: 10