From 5de6a1913bed2bd61ab79737bdc8e41317bf95f2 Mon Sep 17 00:00:00 2001 From: zhupengyang <1165938320@qq.com> Date: Tue, 7 Jan 2020 12:16:38 +0800 Subject: [PATCH] fix test_bilinear_tensor_product_op timeout (#22120) --- .../tests/unittests/test_bilinear_tensor_product_op.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py b/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py index 424239442f9..0db669bc1f9 100644 --- a/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py +++ b/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py @@ -23,9 +23,9 @@ class TestBilinearTensorProductOp(OpTest): def setUp(self): self.op_type = "bilinear_tensor_product" batch_size = 6 - size0 = 30 - size1 = 20 - size2 = 100 + size0 = 5 + size1 = 4 + size2 = 5 a = np.random.random((batch_size, size0)).astype("float64") b = np.random.random((batch_size, size1)).astype("float64") w = np.random.random((size2, size0, size1)).astype("float64") -- GitLab