add ignore_index for test case

9cdf6bd6 · root · chajchaj · 94952111 · 9cdf6bd6
隐藏空白更改
内联并排

Showing with 8 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py ...n/paddle/fluid/tests/unittests/test_cross_entropy_loss.py +8 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -1375,7 +1375,10 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                label_data[0] = 255
                weight_data = paddle.rand([100])
                paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data, weight=weight_data)
+                    input=input_data,
+                    label=label_data,
+                    weight=weight_data,
+                    ignore_index=0)

            self.assertRaises(ValueError, test_LabelValue)

@@ -1386,7 +1389,10 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                label_data[0] = -1
                weight_data = paddle.rand([100])
                paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data, weight=weight_data)
+                    input=input_data,
+                    label=label_data,
+                    weight=weight_data,
+                    ignore_index=0)

            self.assertRaises(ValueError, test_LabelValueNeg)