diff --git a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py index 01710d579a6fbb0d7b15588767b8865bf95b4609..731e4b54e22c35ceaf056431429e19f4f47993cb 100644 --- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py +++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py @@ -1378,7 +1378,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): input=input_data, label=label_data, weight=weight_data, - ignore_index=0) + ignore_index=255) self.assertRaises(ValueError, test_LabelValue) @@ -1392,7 +1392,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase): input=input_data, label=label_data, weight=weight_data, - ignore_index=0) + ignore_index=-1) self.assertRaises(ValueError, test_LabelValueNeg)