transformer loss fix.

4 years ago · bab4757904
parent ed38d787d1
commit bab4757904
1 changed files with 1 additions and 1 deletions
--- 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,