Skip to content

P
Paddle

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

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 7afd7f3d 编写于 作者: H HydrogenSulfate 提交者： chajchaj
浏览文件
操作

Update test_cross_entropy_loss.py

上级 f6dc4b6b
显示空白变更内容
内联 并排
Showing with 2 addition and 4 deletion
python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
浏览文件 @ 7afd7f3d
......@@ -851,8 +851,7 @@ class CrossEntropyLoss(unittest.TestCase):
with fluid.dygraph.guard():
cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
weight=fluid.dygraph.to_variable(weight_np),
ignore_index=255)
weight=fluid.dygraph.to_variable(weight_np), ignore_index=255)
dy_ret = cross_entropy_loss(
fluid.dygraph.to_variable(input_np),
fluid.dygraph.to_variable(label_np))
......@@ -1189,8 +1188,7 @@ class CrossEntropyLoss(unittest.TestCase):
weight_np = np.random.random([C]).astype(self.dtype)
with fluid.dygraph.guard():
cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
weight=fluid.dygraph.to_variable(weight_np),
ignore_index=255)
weight=fluid.dygraph.to_variable(weight_np), ignore_index=255)
dy_ret = cross_entropy_loss(
fluid.dygraph.to_variable(input_np),
fluid.dygraph.to_variable(label_np))
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或