From 662da12c1c3a31540745c8b4f3ad448dfa33a678 Mon Sep 17 00:00:00 2001 From: liuhongyu Date: Mon, 3 Dec 2018 15:06:36 +0800 Subject: [PATCH] fix log --- fluid/PaddleNLP/language_model/lstm/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fluid/PaddleNLP/language_model/lstm/train.py b/fluid/PaddleNLP/language_model/lstm/train.py index b4762d5e..2599a92c 100644 --- a/fluid/PaddleNLP/language_model/lstm/train.py +++ b/fluid/PaddleNLP/language_model/lstm/train.py @@ -286,8 +286,9 @@ def train(): print("train ppl", ppl[0]) if epoch_id == max_epoch - 1 and args.enable_ce: - print("lstm_language_model_duration\t%s" % (total_time / max_epoch)) - print("lstm_language_model_loss\t%s" % ppl[0]) + print("ptblm\tlstm_language_model_duration\t%s" % + (total_time / max_epoch)) + print("ptblm\tlstm_language_model_loss\t%s" % ppl[0]) model_path = os.path.join("model_new/", str(epoch_id)) if not os.path.isdir(model_path): -- GitLab