Skip to content

Commit 86bc4a6

Browse files
authored
Merge pull request #1849 from reyoung/correct_model_average_usage
Correct model average settings in crf
2 parents 2cef071 + 07de602 commit 86bc4a6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

demo/sequence_tagging/linear_crf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
learning_method=MomentumOptimizer(),
2828
batch_size=batch_size,
2929
regularization=L2Regularization(batch_size * 1e-4),
30-
average_window=0.5,
30+
model_average=ModelAverage(0.5),
3131
learning_rate=1e-1,
3232
learning_rate_decay_a=1e-5,
3333
learning_rate_decay_b=0.25, )

demo/sequence_tagging/rnn_crf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
learning_method=MomentumOptimizer(),
2828
batch_size=batch_size,
2929
regularization=L2Regularization(batch_size * 1e-5),
30-
average_window=0.5,
30+
model_average=ModelAverage(0.5),
3131
learning_rate=2e-3,
3232
learning_rate_decay_a=5e-7,
3333
learning_rate_decay_b=0.5, )

0 commit comments

Comments
 (0)