diff --git a/python/paddle/fluid/tests/unittests/test_smooth_l1_loss_op.py b/python/paddle/fluid/tests/unittests/test_smooth_l1_loss_op.py index 201aa17cdd..217bae7c06 100644 --- a/python/paddle/fluid/tests/unittests/test_smooth_l1_loss_op.py +++ b/python/paddle/fluid/tests/unittests/test_smooth_l1_loss_op.py @@ -64,7 +64,7 @@ class TestSmoothL1LossOp1(OpTest): class TestSmoothL1LossOp2(OpTest): def setUp(self): self.op_type = "smooth_l1_loss" - dims = (5, 10) + dims = (5, 20) self.inputs = { 'X': np.random.random(dims).astype("float32"), 'Y': np.random.random(dims).astype("float32"), @@ -88,7 +88,7 @@ class TestSmoothL1LossOp2(OpTest): self.check_output() def test_check_grad_normal(self): - self.check_grad(['X', 'Y'], 'Out') + self.check_grad(['X', 'Y'], 'Out', max_relative_error=0.03) def test_check_grad_ingore_x(self): self.check_grad( diff --git a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py index 317abe124e..2a2f7c84fb 100644 --- a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py +++ b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py @@ -26,7 +26,6 @@ NEED_TO_FIX_OP_LIST = [ 'matmul', 'mul', 'scatter', - 'smooth_l1_loss', 'soft_relu', 'squared_l2_distance', 'tree_conv',