diff --git a/demo/sequence_tagging/linear_crf.py b/demo/sequence_tagging/linear_crf.py index 0624b17787aaf90732707e5f4fc6c2195b8f65ee..ea012ba1ae9c790ccefd3dd5f066aa92202128a2 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 b9b41b2433461eb1bfb309659834661c2ae43253..937a34df103663ecf0f0827bbfb9d82823c9b902 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, )