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 4af5f524a5eedd7af6f30f9f12678120867f5aa0..bffb4f3b666a7ddcc133b7c30fab132b49aa1d0e 100644 --- a/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py @@ -84,18 +84,11 @@ class TestCase2(TestBilinearInterpOp): self.out_w = 12 -class TestCase2(TestBilinearInterpOp): - def init_test_case(self): - self.input_shape = [16, 3, 512, 1024] - self.out_h = 128 - self.out_w = 256 - - -class TestCase2(TestBilinearInterpOp): +class TestCase3(TestBilinearInterpOp): def init_test_case(self): - self.input_shape = [8, 1, 256, 128] - self.out_h = 1024 - self.out_w = 1024 + self.input_shape = [1, 1, 128, 64] + self.out_h = 64 + self.out_w = 128 if __name__ == "__main__":