From c33cc83e7a321c79b5dae38aa32145647a32f728 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Sun, 21 Jun 2015 17:22:01 -0400 Subject: Fix candidate_encoder dim_input --- 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 index 00fc958..36d16ed 100644 --- 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) -- cgit v1.2.3