diff --git a/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py b/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py index 385c18df77cbf1440928aac5428b7d3f6c4d37ee..73b11de069c9026cb5449b72543bc9087deb6703 100755 --- a/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py @@ -178,9 +178,9 @@ class TestBilinearInterpCase2(TestBilinearInterpOp): class TestBilinearInterpCase3(TestBilinearInterpOp): def init_test_case(self): self.interp_method = 'bilinear' - self.input_shape = [1, 1, 128, 64] + self.input_shape = [1, 1, 32, 64] self.out_h = 64 - self.out_w = 128 + self.out_w = 32 self.scale = 0. self.align_corners = True self.align_mode = 1 @@ -213,11 +213,11 @@ class TestBilinearInterpCase5(TestBilinearInterpOp): class TestBilinearInterpCase6(TestBilinearInterpOp): def init_test_case(self): self.interp_method = 'bilinear' - self.input_shape = [1, 1, 128, 64] + self.input_shape = [1, 1, 32, 64] self.out_h = 64 - self.out_w = 128 + self.out_w = 32 self.scale = 0. - self.out_size = np.array([65, 129]).astype("int32") + self.out_size = np.array([65, 33]).astype("int32") self.align_corners = True self.align_mode = 1 @@ -225,8 +225,8 @@ class TestBilinearInterpCase6(TestBilinearInterpOp): class TestBilinearInterpSame(TestBilinearInterpOp): def init_test_case(self): self.interp_method = 'bilinear' - self.input_shape = [2, 3, 128, 64] - self.out_h = 128 + self.input_shape = [2, 3, 32, 64] + self.out_h = 32 self.out_w = 64 self.scale = 0. self.align_corners = True @@ -307,9 +307,9 @@ class TestBilinearInterpOpUint8(OpTest): class TestBilinearInterpCase1Uint8(TestBilinearInterpOpUint8): def init_test_case(self): self.interp_method = 'bilinear' - self.input_shape = [2, 3, 128, 64] - self.out_h = 120 - self.out_w = 50 + self.input_shape = [2, 3, 32, 64] + self.out_h = 64 + self.out_w = 32 self.scale = 0. self.align_corners = True self.align_mode = 1 diff --git a/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py b/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py index 153a1073d0946438963413cccd430d8dc3908714..a8d16677e22127a484e1c6987d7f6d47fdc220bd 100755 --- a/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py @@ -152,9 +152,9 @@ class TestNearestNeighborInterpCase2(TestNearestInterpOp): class TestNearestNeighborInterpCase3(TestNearestInterpOp): def init_test_case(self): self.interp_method = 'nearest' - self.input_shape = [1, 1, 128, 64] + self.input_shape = [1, 1, 32, 64] self.out_h = 64 - self.out_w = 128 + self.out_w = 32 self.scale = 0. self.align_corners = True @@ -184,9 +184,9 @@ class TestNearestNeighborInterpCase5(TestNearestInterpOp): class TestNearestNeighborInterpCase6(TestNearestInterpOp): def init_test_case(self): self.interp_method = 'nearest' - self.input_shape = [1, 1, 128, 64] + self.input_shape = [1, 1, 32, 64] self.out_h = 64 - self.out_w = 128 + self.out_w = 32 self.scale = 0. self.out_size = np.array([65, 129]).astype("int32") self.align_corners = True @@ -195,8 +195,8 @@ class TestNearestNeighborInterpCase6(TestNearestInterpOp): class TestNearestNeighborInterpSame(TestNearestInterpOp): def init_test_case(self): self.interp_method = 'nearest' - self.input_shape = [2, 3, 128, 64] - self.out_h = 128 + self.input_shape = [2, 3, 32, 64] + self.out_h = 32 self.out_w = 64 self.scale = 0. self.align_corners = True @@ -271,9 +271,9 @@ class TestNearestInterpOpUint8(OpTest): class TestNearestNeighborInterpCase1Uint8(TestNearestInterpOpUint8): def init_test_case(self): self.interp_method = 'nearest' - self.input_shape = [2, 3, 128, 64] - self.out_h = 120 - self.out_w = 50 + self.input_shape = [2, 3, 32, 64] + self.out_h = 80 + self.out_w = 40 self.scale = 0. self.align_corners = True diff --git a/python/paddle/fluid/tests/unittests/test_trilinear_interp_op.py b/python/paddle/fluid/tests/unittests/test_trilinear_interp_op.py index 9a0d1bb6bd34b47b5761ae9e12d2c915aaa56c40..40655c0f3cbb5e5267bf33a1ad935b2a5f64af47 100755 --- a/python/paddle/fluid/tests/unittests/test_trilinear_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_trilinear_interp_op.py @@ -428,8 +428,8 @@ class TestTrilinearInterpScale2(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_d = 60 + self.out_h = 40 self.out_w = 25 self.scale = 1. self.align_corners = True @@ -440,8 +440,8 @@ class TestTrilinearInterpScale3(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_d = 60 + self.out_h = 40 self.out_w = 25 self.scale = 1.5 self.align_corners = True @@ -452,8 +452,8 @@ class TestTrilinearInterpZero(TestTrilinearInterpOp): def init_test_case(self): self.interp_method = 'trilinear' self.input_shape = [2, 3, 5, 7, 11] - self.out_d = 82 - self.out_h = 60 + self.out_d = 60 + self.out_h = 40 self.out_w = 25 self.scale = 0.2 self.align_corners = False