提交 f95d3f21 编写于 作者: L Li Hongzhang

fix assertion: Tensor(0) is falsy

上级 18253952
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册