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

!8085 Loss saved does not reduce precision in SummaryCollector 

Merge pull request !8085 from ougongchang/fix_round
This commit is contained in:
mindspore-ci-bot 2020-11-02 09:44:37 +08:00 committed by Gitee
parent b6ed32cd0d 87bf9fbb40
commit 74886f790d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/train/callback/_summary_collector.py
View File

@ -548,8 +548,7 @@ class SummaryCollector(Callback):
if not isinstance(loss, Tensor): if not isinstance(loss, Tensor):
loss = Tensor(loss) loss = Tensor(loss)
precision = 4 loss = Tensor(np.mean(loss.asnumpy()))
loss = Tensor(round(np.mean(loss.asnumpy()), precision))
return loss return loss
def _get_optimizer(self, cb_params): def _get_optimizer(self, cb_params):