sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6066 transformer loss fix. 

Merge pull request !6066 from linqingke/fasterrcnn
This commit is contained in:
mindspore-ci-bot 2020-09-12 14:24:36 +08:00 committed by Gitee
parent acc95fc8f6 bab4757904
commit 1bcffd29dd
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,