diff --git a/fluid/PaddleNLP/language_model/lstm/train.py b/fluid/PaddleNLP/language_model/lstm/train.py index b4762d5ec3b85c4a9b99935bce5a1e2efcb755a4..2599a92c7a49af28776bbd2e4886c2009e07fa9f 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):