From bab4757904f06a40e7b8a13b4833003b83f4d423 Mon Sep 17 00:00:00 2001 From: linqingke Date: Fri, 11 Sep 2020 17:00:25 +0800 Subject: [PATCH] transformer loss fix. --- model_zoo/official/nlp/transformer/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/nlp/transformer/train.py b/model_zoo/official/nlp/transformer/train.py index 60a8535699c..6fe89453d02 100644 --- a/model_zoo/official/nlp/transformer/train.py +++ b/model_zoo/official/nlp/transformer/train.py @@ -72,7 +72,7 @@ class LossCallBack(Callback): print("time: {}, epoch: {}, step: {}, outputs are {}".format(time_stamp_current - time_stamp_first, cb_params.cur_epoch_num, cb_params.cur_step_num, str(cb_params.net_outputs))) - with open("./loss_{}.log".fromat(self.rank_id), "a+") as f: + with open("./loss_{}.log".format(self.rank_id), "a+") as f: f.write("time: {}, epoch: {}, step: {}, outputs are {}".format(time_stamp_current - time_stamp_first, cb_params.cur_epoch_num, cb_params.cur_step_num,