From 985e4bae5e41ee2fcce56dd732ae5c93193242b0 Mon Sep 17 00:00:00 2001 From: dyning Date: Thu, 2 Jan 2020 16:38:45 +0800 Subject: [PATCH] fix unittest for spp op, test=develop (#22030) fix unittest for spp op --- python/paddle/fluid/tests/unittests/test_spp_op.py | 7 +++---- .../tests/unittests/white_list/op_accuracy_white_list.py | 6 +++--- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_spp_op.py b/python/paddle/fluid/tests/unittests/test_spp_op.py index ad077236c8..1a134b58de 100644 --- a/python/paddle/fluid/tests/unittests/test_spp_op.py +++ b/python/paddle/fluid/tests/unittests/test_spp_op.py @@ -62,11 +62,10 @@ class TestSppOp(OpTest): self.check_output() def test_check_grad(self): - if self.pool_type != "avg": - self.check_grad(['X'], 'Out', max_relative_error=0.05) + self.check_grad(['X'], 'Out') def init_test_case(self): - self.shape = [4, 2, 4, 4] + self.shape = [3, 2, 4, 4] self.pyramid_height = 3 self.pool2D_forward_naive = max_pool2D_forward_naive self.pool_type = "max" @@ -74,7 +73,7 @@ class TestSppOp(OpTest): class TestCase2(TestSppOp): def init_test_case(self): - self.shape = [3, 2, 4, 4] + self.shape = [3, 2, 16, 16] self.pyramid_height = 3 self.pool2D_forward_naive = avg_pool2D_forward_naive self.pool_type = "avg" diff --git a/python/paddle/fluid/tests/unittests/white_list/op_accuracy_white_list.py b/python/paddle/fluid/tests/unittests/white_list/op_accuracy_white_list.py index ca8821e143..3490643014 100644 --- a/python/paddle/fluid/tests/unittests/white_list/op_accuracy_white_list.py +++ b/python/paddle/fluid/tests/unittests/white_list/op_accuracy_white_list.py @@ -28,9 +28,9 @@ NO_FP64_CHECK_GRAD_OP_LIST = [ 'reduce_min', 'relu', 'reshape2', 'roi_perspective_transform', 'row_conv', 'scale', 'scatter', 'sequence_conv', 'sequence_pool', 'sequence_reverse', 'sequence_slice', 'sequence_topk_avg_pooling', 'shuffle_channel', 'sigmoid', - 'smooth_l1_loss', 'softmax', 'spectral_norm', 'spp', 'sqrt', - 'squared_l2_distance', 'squared_l2_norm', 'tanh', 'transpose2', - 'trilinear_interp', 'var_conv_2d', 'warpctc' + 'smooth_l1_loss', 'softmax', 'spectral_norm', 'sqrt', 'squared_l2_distance', + 'squared_l2_norm', 'tanh', 'transpose2', 'trilinear_interp', 'var_conv_2d', + 'warpctc' ] NO_FP16_CHECK_GRAD_OP_LIST = [ -- GitLab