aboutsummaryrefslogtreecommitdiff
path: root/config/bidirectional_tgtcls_1_momentum.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-07-25 16:21:08 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-07-25 16:21:08 -0400
commit8876175ad46d43e1a387b11ad398defac588b901 (patch)
treed498a41c6745d6300dcf858b6f546988696f5c2f /config/bidirectional_tgtcls_1_momentum.py
parent4b0c1a83bc8028983eeb4c4bda19e05954cb6ac2 (diff)
downloadtaxi-8876175ad46d43e1a387b11ad398defac588b901.tar.gz
taxi-8876175ad46d43e1a387b11ad398defac588b901.zip
Add memory_net_bidir_mom
Diffstat (limited to 'config/bidirectional_tgtcls_1_momentum.py')
-rw-r--r--config/bidirectional_tgtcls_1_momentum.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config/bidirectional_tgtcls_1_momentum.py b/config/bidirectional_tgtcls_1_momentum.py
index b286e0f..65ad021 100644
--- a/config/bidirectional_tgtcls_1_momentum.py
+++ b/config/bidirectional_tgtcls_1_momentum.py
@@ -27,12 +27,12 @@ embed_weights_init = IsotropicGaussian(0.01)
weights_init = IsotropicGaussian(0.1)
biases_init = Constant(0.01)
-batch_size = 100
+batch_size = 300
batch_sort_size = 20
max_splits = 100
# monitor_freq = 10000 # temporary, for finding good learning rate
-step_rule= Momentum(learning_rate=0.1, momentum=0.9)
+step_rule= Momentum(learning_rate=0.01, momentum=0.9)