diff options
| -rw-r--r-- | config/memory_network_1.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |