From f95d3f21fb06541ead74cbea517525d9d1396b12 Mon Sep 17 00:00:00 2001 From: Li Hongzhang Date: Sat, 29 Aug 2020 15:43:20 +0800 Subject: [PATCH] fix assertion: Tensor(0) is falsy --- mindspore/train/callback/_summary_collector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/callback/_summary_collector.py b/mindspore/train/callback/_summary_collector.py index d4f6da8423..63706ca73e 100644 --- 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