diff options
Diffstat (limited to 'config/memory_network_mlp_3_momentum.py')
| -rw-r--r-- | config/memory_network_mlp_3_momentum.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config/memory_network_mlp_3_momentum.py b/config/memory_network_mlp_3_momentum.py index 241142e..83fbc96 100644 --- a/config/memory_network_mlp_3_momentum.py +++ b/config/memory_network_mlp_3_momentum.py @@ -41,9 +41,9 @@ candidate_encoder.dim_embeddings = dim_embeddings representation_size = 500 representation_activation = Tanh -normalize_representation = True +normalize_representation = False -step_rule = Momentum(learning_rate=0.01, momentum=0.9) +step_rule = Momentum(learning_rate=0.001, momentum=0.9) batch_size = 5000 # batch_sort_size = 20 |