commit c33cc83e7a321c79b5dae38aa32145647a32f728 parent 2a6980fdac3f6c3987d92882368bf413b50dee36 Author: Étienne Simon <esimon@esimon.eu> Date: Sun, 21 Jun 2015 17:22:01 -0400 Fix candidate_encoder dim_input Diffstat:
| M | config/memory_network_1.py | | | 2 | +- |
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/config/memory_network_1.py b/config/memory_network_1.py @@ -26,7 +26,7 @@ prefix_encoder.weights_init = IsotropicGaussian(0.01) prefix_encoder.biases_init = Constant(0.001) candidate_encoder = MLPConfig() -candidate_encoder.dim_input = n_begin_end_pts * 2 + sum(x for (_, _, x) in dim_embeddings) +candidate_encoder.dim_input = n_begin_end_pts * 2 * 2 + sum(x for (_, _, x) in dim_embeddings) candidate_encoder.dim_hidden = [50] candidate_encoder.weights_init = IsotropicGaussian(0.01) candidate_encoder.biases_init = Constant(0.001)