diff options
Diffstat (limited to 'config')
| -rw-r--r-- | config/model_0.py | 4 |
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 |