Browse Source

!5501 Fix: Tensor(0) is surprisingly to be falsy, which leads to loss not be recorded

Merge pull request !5501 from LiHongzhang/loss_tensor
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
fdbd4cd4d1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/callback/_summary_collector.py

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

@@ -602,7 +602,7 @@ class SummaryCollector(Callback):
return
train_lineage = {}
loss = self._get_loss(cb_params)
if loss:
if loss is not None:
loss_numpy = loss.asnumpy()
loss = float(np.atleast_1d(loss_numpy)[0])
train_lineage[LineageMetadata.loss] = loss


Loading…
Cancel
Save