diff options
Diffstat (limited to 'config/bidirectional_tgtcls_1_momentum.py')
-rw-r--r-- | config/bidirectional_tgtcls_1_momentum.py | 4 |
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) |