From d62832718d4b80719c26aedbd8f99fc52be13d79 Mon Sep 17 00:00:00 2001 From: Qingsheng Li Date: Wed, 8 Aug 2018 17:59:27 +0800 Subject: [PATCH] Fix seq2seq CE (#1110) * Seq2seq CE: do not skip the first 2 passes --- fluid/neural_machine_translation/rnn_search/_ce.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fluid/neural_machine_translation/rnn_search/_ce.py b/fluid/neural_machine_translation/rnn_search/_ce.py index 32b13c66..e948336e 100644 --- a/fluid/neural_machine_translation/rnn_search/_ce.py +++ b/fluid/neural_machine_translation/rnn_search/_ce.py @@ -7,9 +7,9 @@ from kpi import CostKpi, DurationKpi, AccKpi #### NOTE kpi.py should shared in models in some way!!!! -train_cost_kpi = CostKpi('train_cost', 0.02, actived=True) -test_cost_kpi = CostKpi('test_cost', 0.005, actived=True) -train_duration_kpi = DurationKpi('train_duration', 0.06, actived=True) +train_cost_kpi = CostKpi('train_cost', 0.02, 0, actived=True) +test_cost_kpi = CostKpi('test_cost', 0.005, 0, actived=True) +train_duration_kpi = DurationKpi('train_duration', 0.06, 0, actived=True) tracking_kpis = [ train_cost_kpi, -- GitLab