From 163a3ddfe3ed49556c976b1bf36016104c707845 Mon Sep 17 00:00:00 2001 From: Double_V Date: Mon, 24 Feb 2020 15:12:07 +0800 Subject: [PATCH] fix row_conv_op small size unittest cases (#22606) --- python/paddle/fluid/tests/unittests/test_row_conv_op.py | 8 ++++---- .../tests/unittests/white_list/check_shape_white_list.py | 1 - 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_row_conv_op.py b/python/paddle/fluid/tests/unittests/test_row_conv_op.py index 602f650a50..59caa25405 100644 --- a/python/paddle/fluid/tests/unittests/test_row_conv_op.py +++ b/python/paddle/fluid/tests/unittests/test_row_conv_op.py @@ -50,7 +50,7 @@ class TestRowConvOp1(OpTest): lod = [[2, 3, 2]] T = sum(lod[0]) D = 16 - context_length = 2 + context_length = 8 x = np.random.random((T, D)).astype("float32") wt = np.random.random((context_length, D)).astype("float32") @@ -139,11 +139,11 @@ def row_conv_foward_Tensor(x, wt): class TestRowOpWithTensorInput(OpTest): def setUp(self): self.op_type = "row_conv" - length = [3, 2, 4] + length = [1, 2, 3] B = 2 T = sum(length) - D = 16 - context_length = 2 + D = 20 + context_length = 6 x = np.random.random((B, T, D)).astype("float32") wt = np.random.random((context_length, D)).astype("float32") diff --git a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py index 7cc5445e02..9061daa0fe 100644 --- a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py +++ b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py @@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [ 'margin_rank_loss', 'matmul', 'mul', - 'row_conv', 'scatter', 'smooth_l1_loss', 'soft_relu', -- GitLab