From 07de6022212ffda1dbf73dc62bb753f76697bfb8 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Fri, 21 Apr 2017 19:14:12 +0800 Subject: [PATCH] Correct model average settings in crf --- demo/sequence_tagging/linear_crf.py | 2 +- demo/sequence_tagging/rnn_crf.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/sequence_tagging/linear_crf.py b/demo/sequence_tagging/linear_crf.py index 0624b17787a..ea012ba1ae9 100644 --- a/demo/sequence_tagging/linear_crf.py +++ b/demo/sequence_tagging/linear_crf.py @@ -27,7 +27,7 @@ settings( learning_method=MomentumOptimizer(), batch_size=batch_size, regularization=L2Regularization(batch_size * 1e-4), - average_window=0.5, + model_average=ModelAverage(0.5), learning_rate=1e-1, learning_rate_decay_a=1e-5, learning_rate_decay_b=0.25, ) diff --git a/demo/sequence_tagging/rnn_crf.py b/demo/sequence_tagging/rnn_crf.py index b9b41b24334..937a34df103 100644 --- a/demo/sequence_tagging/rnn_crf.py +++ b/demo/sequence_tagging/rnn_crf.py @@ -27,7 +27,7 @@ settings( learning_method=MomentumOptimizer(), batch_size=batch_size, regularization=L2Regularization(batch_size * 1e-5), - average_window=0.5, + model_average=ModelAverage(0.5), learning_rate=2e-3, learning_rate_decay_a=5e-7, learning_rate_decay_b=0.5, ) -- GitLab