!8085 Loss saved does not reduce precision in SummaryCollector

Merge pull request !8085 from ougongchang/fix_round
2020-11-02 09:44:37 +08:00 · 2020-11-02 09:44:37 +08:00 · 74886f790d
parent b6ed32cd0d 87bf9fbb40
commit 74886f790d
1 changed files with 1 additions and 2 deletions
--- a/mindspore/train/callback/_summary_collector.py
+++ b/mindspore/train/callback/_summary_collector.py
@ -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):