fix assertion: Tensor(0) is falsy

5 years ago · f95d3f21fb
parent 18253952f5
commit f95d3f21fb
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