Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

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

Update test_cross_entropy_loss.py

上级 f2df33e3
隐藏空白更改
内联 并排
Showing with 0 addition and 2 deletion
python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
浏览文件 @ f6dc4b6b
...@@ -852,7 +852,6 @@ class CrossEntropyLoss(unittest.TestCase): ...@@ -852,7 +852,6 @@ class CrossEntropyLoss(unittest.TestCase):
with fluid.dygraph.guard(): with fluid.dygraph.guard():
cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss( cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
weight=fluid.dygraph.to_variable(weight_np), weight=fluid.dygraph.to_variable(weight_np),
axis=1,
ignore_index=255) ignore_index=255)
dy_ret = cross_entropy_loss( dy_ret = cross_entropy_loss(
fluid.dygraph.to_variable(input_np), fluid.dygraph.to_variable(input_np),
...@@ -1191,7 +1190,6 @@ class CrossEntropyLoss(unittest.TestCase): ...@@ -1191,7 +1190,6 @@ class CrossEntropyLoss(unittest.TestCase):
with fluid.dygraph.guard(): with fluid.dygraph.guard():
cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss( cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
weight=fluid.dygraph.to_variable(weight_np), weight=fluid.dygraph.to_variable(weight_np),
axis=1,
ignore_index=255) ignore_index=255)
dy_ret = cross_entropy_loss( dy_ret = cross_entropy_loss(
fluid.dygraph.to_variable(input_np), fluid.dygraph.to_variable(input_np),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或