aboutsummaryrefslogtreecommitdiff
path: root/config/model_0.py
diff options
context:
space:
mode:
Diffstat (limited to 'config/model_0.py')
-rw-r--r--config/model_0.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config/model_0.py b/config/model_0.py
index b157faa..c4985b2 100644
--- a/config/model_0.py
+++ b/config/model_0.py
@@ -12,6 +12,6 @@ dim_input = n_begin_end_pts * 2 * 2 + dim_embed + dim_embed
dim_hidden = [200, 100]
dim_output = 2
-learning_rate = 0.002
-momentum = 0.9
+learning_rate = 0.0001
+momentum = 0.99
batch_size = 32