From 858963b421c229b1420f7cc6d20899aa1285614e Mon Sep 17 00:00:00 2001 From: wangxiao Date: Wed, 4 Dec 2019 16:40:15 +0800 Subject: [PATCH] fix a bug --- paddlepalm/task_paradigm/match.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddlepalm/task_paradigm/match.py b/paddlepalm/task_paradigm/match.py index d42c64a..df4e43b 100644 --- a/paddlepalm/task_paradigm/match.py +++ b/paddlepalm/task_paradigm/match.py @@ -79,8 +79,8 @@ class TaskParadigm(task_paradigm): initializer=fluid.initializer.Constant(0.))) if self._is_training: - ce_loss, probs = fluid.layers.cross_entropy( - input=logits, label=labels, return_softmax=True) + ce_loss = fluid.layers.cross_entropy( + input=logits, label=labels) loss = fluid.layers.mean(x=ce_loss) return {'loss': loss} else: -- GitLab