aboutsummaryrefslogtreecommitdiff
path: root/config/bidirectional_tgtcls_1_momentum.py
diff options
context:
space:
mode:
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 65ad021..99b50d0 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 = 300
+batch_size = 200
batch_sort_size = 20
max_splits = 100
# monitor_freq = 10000 # temporary, for finding good learning rate
-step_rule= Momentum(learning_rate=0.01, momentum=0.9)
+step_rule= Momentum(learning_rate=0.001, momentum=0.9)