Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 94952111 编写于 作者: R root 提交者： chajchaj
浏览文件
操作

add weigth data to unit test

上级 e1ea895f
隐藏空白更改
内联 并排
Showing with 4 addition and 2 deletion
python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
浏览文件 @ 94952111
...@@ -1373,8 +1373,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase): ...@@ -1373,8 +1373,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
label_data = paddle.randint( label_data = paddle.randint(
0, 100, shape=[20, 1], dtype="int64") 0, 100, shape=[20, 1], dtype="int64")
label_data[0] = 255 label_data[0] = 255
weight_data = paddle.rand([100])
paddle.nn.functional.cross_entropy( paddle.nn.functional.cross_entropy(
input=input_data, label=label_data) input=input_data, label=label_data, weight=weight_data)
self.assertRaises(ValueError, test_LabelValue) self.assertRaises(ValueError, test_LabelValue)
...@@ -1383,8 +1384,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase): ...@@ -1383,8 +1384,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
label_data = paddle.randint( label_data = paddle.randint(
0, 100, shape=[20, 1], dtype="int64") 0, 100, shape=[20, 1], dtype="int64")
label_data[0] = -1 label_data[0] = -1
weight_data = paddle.rand([100])
paddle.nn.functional.cross_entropy( paddle.nn.functional.cross_entropy(
input=input_data, label=label_data) input=input_data, label=label_data, weight=weight_data)
self.assertRaises(ValueError, test_LabelValueNeg) self.assertRaises(ValueError, test_LabelValueNeg)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或