diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-25 17:31:17 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-25 17:31:17 -0400 |
| commit | 2a20bc827a8c1c9b6e74ef4e1234788207be45b8 (patch) | |
| tree | 2051f9f81324bbf202a3bc3fa8c59a68288d2566 /config/memory_network_bidir_momentum.py | |
| parent | d3dbbe68642320b9a225b1d3515f0181916df8ad (diff) | |
| download | taxi-2a20bc827a8c1c9b6e74ef4e1234788207be45b8.tar.gz taxi-2a20bc827a8c1c9b6e74ef4e1234788207be45b8.zip | |
Add batch sorting
Diffstat (limited to 'config/memory_network_bidir_momentum.py')
| -rw-r--r-- | config/memory_network_bidir_momentum.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/config/memory_network_bidir_momentum.py b/config/memory_network_bidir_momentum.py index e5863ae..3d90494 100644 --- a/config/memory_network_bidir_momentum.py +++ b/config/memory_network_bidir_momentum.py @@ -51,8 +51,8 @@ batch_sort_size = 20 max_splits = 100 num_cuts = 1000 -train_candidate_size = 1000 -valid_candidate_size = 1000 -test_candidate_size = 1000 +train_candidate_size = 300 +valid_candidate_size = 300 +test_candidate_size = 300 step_rule = Momentum(learning_rate=0.01, momentum=0.9) |