Added generator template config

IBM · Nov 16, 2018 · 05f6dd9 · 05f6dd9
1 parent a3bb087
commit 05f6dd9
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/configs/problem_inits/COG_dataset_generator.yaml b/configs/problem_inits/COG_dataset_generator.yaml
@@ -0,0 +1,13 @@
+# Problem parameters:
+training:
+ problem:
+  name: COG
+  data_folder: '~/data/cog'
+  set: train
+  dataset_type: generated
+  generation:
+   examples_per_task: 256
+   sequence_length: 5
+   memory_length: 4
+   max_distractors: 2
+   nr_processors: 6