Loss saved does not reduce precision in SummaryCollector

4 years ago · 87bf9fbb40
parent f7012b8205
commit 87bf9fbb40
1 changed files with 1 additions and 2 deletions
--- a/mindspore/train/callback/_summary_collector.py
+++ b/mindspore/train/callback/_summary_collector.py
@ -543,8 +543,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):