aboutsummaryrefslogtreecommitdiff
path: root/config
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-07-10 17:16:20 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-07-10 17:16:20 -0400
commit27a0e0949c6ca3f7bd18569a23ddd0e1b3e9a64e (patch)
tree1201f8abe7c664f15084b1d48ed8de92c72879d8 /config
parent793be7b049cecba43072858341dc7006fef352e7 (diff)
downloadtaxi-27a0e0949c6ca3f7bd18569a23ddd0e1b3e9a64e.tar.gz
taxi-27a0e0949c6ca3f7bd18569a23ddd0e1b3e9a64e.zip
Batch shuffling
Diffstat (limited to 'config')
-rw-r--r--config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py b/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py
index a4db33c..b816930 100644
--- a/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py
+++ b/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py
@@ -23,14 +23,14 @@ dim_embeddings = [
]
dim_input = n_begin_end_pts * 2 * 2 + sum(x for (_, _, x) in dim_embeddings)
-dim_hidden = [1000]
+dim_hidden = [500]
dim_output = tgtcls.shape[0]
embed_weights_init = IsotropicGaussian(0.01)
mlp_weights_init = IsotropicGaussian(0.1)
mlp_biases_init = Constant(0.01)
-step_rule = Momentum(learning_rate=0.01, momentum=0.9)
+step_rule = Momentum(learning_rate=0.001, momentum=0.99)
batch_size = 200