diff options
author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-28 09:20:24 -0400 |
---|---|---|
committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-28 09:20:24 -0400 |
commit | 1c8241ab7a5e933c8a3452b407a1be054467613b (patch) | |
tree | 53605d2ad6a4c9f0da1d7061e92d7d26abf239f7 | |
parent | 683e8788b4a52dfaf134539283a47810ba2c3420 (diff) | |
download | taxi-1c8241ab7a5e933c8a3452b407a1be054467613b.tar.gz taxi-1c8241ab7a5e933c8a3452b407a1be054467613b.zip |
Memory net 3 candidate size
-rw-r--r-- | config/memory_network_mlp_3_momentum.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/config/memory_network_mlp_3_momentum.py b/config/memory_network_mlp_3_momentum.py index 5e1e4b2..241142e 100644 --- a/config/memory_network_mlp_3_momentum.py +++ b/config/memory_network_mlp_3_momentum.py @@ -45,11 +45,11 @@ normalize_representation = True step_rule = Momentum(learning_rate=0.01, momentum=0.9) -batch_size = 500 +batch_size = 5000 # batch_sort_size = 20 -max_splits = 100 +max_splits = 200 -train_candidate_size = 2000 -valid_candidate_size = 2000 -test_candidate_size = 2000 +train_candidate_size = 10000 +valid_candidate_size = 10000 +test_candidate_size = 10000 |