Browse Source

Loss saved does not reduce precision in SummaryCollector

tags/v1.1.0
ougongchang 5 years ago
parent
commit
87bf9fbb40
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/train/callback/_summary_collector.py

+ 1
- 2
mindspore/train/callback/_summary_collector.py View File

@@ -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):


Loading…
Cancel
Save