Move weights init to config files ; fix s/time/travel_time

1 files changed, 3 insertions, 5 deletions

diff --git a/model/dest_simple_mlp_tgtcls.py b/model/dest_simple_mlp_tgtcls.py
index a7b6f9b..1381d4c 100644
--- a/model/dest_simple_mlp_tgtcls.py
+++ b/model/dest_simple_mlp_tgtcls.py
@@ -6,8 +6,6 @@ from theano import tensor
 from blocks.bricks import MLP, Rectifier, Linear, Sigmoid, Identity, Softmax
 from blocks.bricks.lookup import LookupTable
 
-from blocks.initialization import IsotropicGaussian, Constant
-
 import data
 import error
 
@@ -60,9 +58,9 @@ class Model(object):
 
         # Initialization
         for tbl in embed_tables:
-            tbl.weights_init = IsotropicGaussian(0.001)
-        mlp.weights_init = IsotropicGaussian(0.01)
-        mlp.biases_init = Constant(0.001)
+            tbl.weights_init = config.embed_weights_init
+        mlp.weights_init = config.mlp_weights_init
+        mlp.biases_init = config.mlp_biases_init
 
         for tbl in embed_tables:
             tbl.initialize()