aboutsummaryrefslogtreecommitdiff
path: root/config/memory_network_mlp_2.py
diff options
context:
space:
mode:
Diffstat (limited to 'config/memory_network_mlp_2.py')
-rw-r--r--config/memory_network_mlp_2.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/config/memory_network_mlp_2.py b/config/memory_network_mlp_2.py
index a64a4fc..33b4867 100644
--- a/config/memory_network_mlp_2.py
+++ b/config/memory_network_mlp_2.py
@@ -43,11 +43,11 @@ representation_activation = Tanh
normalize_representation = True
-batch_size = 100
-batch_sort_size = 20
+batch_size = 1000
+# batch_sort_size = 20
max_splits = 100
-train_candidate_size = 1000
-valid_candidate_size = 1000
-test_candidate_size = 1000
+train_candidate_size = 5000
+valid_candidate_size = 5000
+test_candidate_size = 5000