Browse Source

!6066 transformer loss fix.

Merge pull request !6066 from linqingke/fasterrcnn
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
1bcffd29dd
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/nlp/transformer/train.py

+ 1
- 1
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,
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,


Loading…
Cancel
Save