diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-27 12:59:39 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-27 12:59:39 -0400 |
| commit | 0021c3fb99d1cd3f8792a8cf5c35548815536428 (patch) | |
| tree | 9edb909def7652579a1b6a40ecb311a61b40455b /config/memory_network_bidir_momentum.py | |
| parent | 2a20bc827a8c1c9b6e74ef4e1234788207be45b8 (diff) | |
| download | taxi-0021c3fb99d1cd3f8792a8cf5c35548815536428.tar.gz taxi-0021c3fb99d1cd3f8792a8cf5c35548815536428.zip | |
Config files
Diffstat (limited to 'config/memory_network_bidir_momentum.py')
| -rw-r--r-- | config/memory_network_bidir_momentum.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/config/memory_network_bidir_momentum.py b/config/memory_network_bidir_momentum.py index 3d90494..504fdee 100644 --- a/config/memory_network_bidir_momentum.py +++ b/config/memory_network_bidir_momentum.py @@ -45,14 +45,14 @@ representation_activation = Tanh normalize_representation = True -batch_size = 32 +batch_size = 64 batch_sort_size = 20 max_splits = 100 num_cuts = 1000 -train_candidate_size = 300 -valid_candidate_size = 300 -test_candidate_size = 300 +train_candidate_size = 100 +valid_candidate_size = 100 +test_candidate_size = 100 step_rule = Momentum(learning_rate=0.01, momentum=0.9) |