!8085 Loss saved does not reduce precision in SummaryCollector

Merge pull request !8085 from ougongchang/fix_round
pull/6890/head
mindspore-ci-bot 4 years ago committed by Gitee
commit 74886f790d

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

Loading…
Cancel
Save