From 87bf9fbb40bc671011b4546725f9e3b758b74ee3 Mon Sep 17 00:00:00 2001 From: ougongchang Date: Sat, 31 Oct 2020 17:11:19 +0800 Subject: [PATCH] Loss saved does not reduce precision in SummaryCollector --- mindspore/train/callback/_summary_collector.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index 630c047fcc..4765deca94 100644 --- 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):