diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-06-21 17:22:01 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-06-21 17:22:01 -0400 |
| commit | c33cc83e7a321c79b5dae38aa32145647a32f728 (patch) | |
| tree | 9d302b0d1d085c39c5c078811f3acb1d2df27027 /config/memory_network_1.py | |
| parent | 2a6980fdac3f6c3987d92882368bf413b50dee36 (diff) | |
| download | taxi-c33cc83e7a321c79b5dae38aa32145647a32f728.tar.gz taxi-c33cc83e7a321c79b5dae38aa32145647a32f728.zip | |
Fix candidate_encoder dim_input
Diffstat (limited to 'config/memory_network_1.py')
| -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) |