Merge pull request PaddlePaddle#1849 from reyoung/correct_model_avera…

…ge_usage Correct model average settings in crf
roysh · Apr 21, 2017 · 86bc4a6 · 86bc4a6
2 parents 2cef071 + 07de602
commit 86bc4a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo/sequence_tagging/linear_crf.py b/demo/sequence_tagging/linear_crf.py
@@ -27,7 +27,7 @@
     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
@@ -27,7 +27,7 @@
     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, )