From 683e8788b4a52dfaf134539283a47810ba2c3420 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Mon, 27 Jul 2015 15:27:44 -0400 Subject: Memory network configurations --- config/memory_network_mlp_2_momentum.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'config/memory_network_mlp_2_momentum.py') diff --git a/config/memory_network_mlp_2_momentum.py b/config/memory_network_mlp_2_momentum.py index 8a0ac81..483dfe8 100644 --- a/config/memory_network_mlp_2_momentum.py +++ b/config/memory_network_mlp_2_momentum.py @@ -45,11 +45,11 @@ normalize_representation = True step_rule = Momentum(learning_rate=0.01, momentum=0.9) -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 -- cgit v1.2.3