From f6dc4b6b3766ee5b06352591757d232eb831f94c Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Sun, 15 Aug 2021 23:39:23 +0800
Subject: [PATCH] Update test_cross_entropy_loss.py

---
 python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py | 2 --
 1 file changed, 2 deletions(-)

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 b6b08c185e4..650c71f4c58 100644
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -852,7 +852,6 @@ class CrossEntropyLoss(unittest.TestCase):
         with fluid.dygraph.guard():
             cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
                 weight=fluid.dygraph.to_variable(weight_np),
-                axis=1,
                 ignore_index=255)
             dy_ret = cross_entropy_loss(
                 fluid.dygraph.to_variable(input_np),
@@ -1191,7 +1190,6 @@ class CrossEntropyLoss(unittest.TestCase):
         with fluid.dygraph.guard():
             cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
                 weight=fluid.dygraph.to_variable(weight_np),
-                axis=1,
                 ignore_index=255)
             dy_ret = cross_entropy_loss(
                 fluid.dygraph.to_variable(input_np),
-- 
GitLab