From 7ddfec009ce30f0aa871b0ee338ac38a5603aec8 Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Mon, 27 Dec 2021 13:11:39 +0800 Subject: [PATCH] change to IndexError --- 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 f13f14cdde..cdf80fb58d 100755 --- a/python/paddle/nn/functional/loss.py +++ b/python/paddle/nn/functional/loss.py @@ -1710,7 +1710,7 @@ def cross_entropy(input, if len(paddle.nonzero(valid_label < 0)) > 0: invalid_label = paddle.gather_nd( valid_label, paddle.nonzero(valid_label < 0)) - raise ValueError( + raise IndexError( "Target({}) is out of class_dimension's lower bound({})". format(invalid_label[0], 0)) # TODO: Temporarily use paddle.nonzero instead of paddle.max @@ -1719,7 +1719,7 @@ def cross_entropy(input, invalid_label = paddle.gather_nd( valid_label, paddle.nonzero(valid_label >= input.shape[axis])) - raise ValueError( + raise IndexError( "Target({}) is out of class_dimension's upper bound({})". format(invalid_label[0], input.shape[axis] - 1)) -- GitLab