diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-27 15:27:44 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-27 15:27:44 -0400 |
| commit | 683e8788b4a52dfaf134539283a47810ba2c3420 (patch) | |
| tree | f711ef1de4e97e0e31269b9e26201805d0f1200a /config/memory_network_mlp_2.py | |
| parent | ff1502ff1b6a4192974f73347b365a5d3a0e1f20 (diff) | |
| download | taxi-683e8788b4a52dfaf134539283a47810ba2c3420.tar.gz taxi-683e8788b4a52dfaf134539283a47810ba2c3420.zip | |
Memory network configurations
Diffstat (limited to 'config/memory_network_mlp_2.py')
| -rw-r--r-- | config/memory_network_mlp_2.py | 10 |
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 |