diff --git a/python/paddle/fluid/tests/unittests/test_trilinear_interp_v2_op.py b/python/paddle/fluid/tests/unittests/test_trilinear_interp_v2_op.py index 245c2623b869af30acfb5d0379c7597813645031..1f8ff4963ec3dc60402f18f332d3402444f4379a 100755 --- a/python/paddle/fluid/tests/unittests/test_trilinear_interp_v2_op.py +++ b/python/paddle/fluid/tests/unittests/test_trilinear_interp_v2_op.py @@ -459,9 +459,9 @@ class TestTrilinearInterpScale1(TestTrilinearInterpOp): def init_test_case(self): self.interp_method = 'trilinear' self.input_shape = [2, 3, 5, 7, 9] - self.out_d = 82 - self.out_h = 60 - self.out_w = 25 + self.out_d = 19 + self.out_h = 15 + self.out_w = 8 self.scale = 2. self.align_corners = True self.align_mode = 1 @@ -471,8 +471,8 @@ class TestTrilinearInterpScale2(TestTrilinearInterpOp): def init_test_case(self): self.interp_method = 'trilinear' self.input_shape = [2, 3, 5, 7, 9] - self.out_d = 60 - self.out_h = 40 + self.out_d = 30 + self.out_h = 20 self.out_w = 25 self.scale = 1. self.align_corners = True @@ -483,8 +483,8 @@ class TestTrilinearInterpScale3(TestTrilinearInterpOp): def init_test_case(self): self.interp_method = 'trilinear' self.input_shape = [2, 3, 5, 7, 9] - self.out_d = 60 - self.out_h = 40 + self.out_d = 30 + self.out_h = 20 self.out_w = 25 self.scale = 1.5 self.align_corners = True @@ -495,8 +495,8 @@ class TestTrilinearInterpZero(TestTrilinearInterpOp): def init_test_case(self): self.interp_method = 'trilinear' self.input_shape = [2, 3, 5, 7, 11] - self.out_d = 60 - self.out_h = 40 + self.out_d = 30 + self.out_h = 20 self.out_w = 25 self.scale = 0.0 self.align_corners = False