From 39e8153223b3f76ee69f452785cf694744f528cc Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Sun, 15 Aug 2021 16:55:30 +0800 Subject: [PATCH] Update loss.py --- python/paddle/nn/functional/loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py index 46ff6bb5fb4..3fd9f45d962 100755 --- a/python/paddle/nn/functional/loss.py +++ b/python/paddle/nn/functional/loss.py @@ -1814,8 +1814,8 @@ def cross_entropy(input, paddle.zeros( [1], dtype=label.dtype), label) - if (len(paddle.nonzero(valid_label < 0)) > 0) or ( - len(paddle.nonzero(valid_label >= input.shape[-1])) > 0): + if (paddle.numel(paddle.nonzero(valid_label < 0)) > 0) or ( + paddle.numel(paddle.nonzero(valid_label >= input.shape[-1])) > 0): invalid_label = paddle.gather_nd( input, paddle.nonzero(valid_label < 0)) if paddle.numel(invalid_label) > 0: -- GitLab