p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

transformer loss fix.

This commit is contained in:
linqingke 2020-09-11 17:00:25 +08:00
parent ed38d787d1
commit bab4757904
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/nlp/transformer/train.py
View File

@ -72,7 +72,7 @@ class LossCallBack(Callback):
print("time: {}, epoch: {}, step: {}, outputs are {}".format(time_stamp_current - time_stamp_first, print("time: {}, epoch: {}, step: {}, outputs are {}".format(time_stamp_current - time_stamp_first,
cb_params.cur_epoch_num, cb_params.cur_step_num, cb_params.cur_epoch_num, cb_params.cur_step_num,
str(cb_params.net_outputs))) 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, f.write("time: {}, epoch: {}, step: {}, outputs are {}".format(time_stamp_current - time_stamp_first,
cb_params.cur_epoch_num, cb_params.cur_epoch_num,
cb_params.cur_step_num, cb_params.cur_step_num,