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

Merge pull request !5501 from LiHongzhang/loss_tensor
2020-08-30 09:52:54 +08:00 · 2020-08-30 09:52:54 +08:00 · fdbd4cd4d1
parent c5bea5575a f95d3f21fb
commit fdbd4cd4d1
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/callback/_summary_collector.py
+++ b/mindspore/train/callback/_summary_collector.py
@ -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