From a06f4b2b2c3dbc7d58744c01a1b429007f107fd0 Mon Sep 17 00:00:00 2001 From: Jiabin Yang Date: Mon, 8 Apr 2019 21:13:53 +0800 Subject: [PATCH] make less batch of tests to fit ci (#16706) * make less batch of tests to fit ci * test=develop, invoke ci add some comments back --- .../fluid/tests/unittests/test_imperative_transformer.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_imperative_transformer.py b/python/paddle/fluid/tests/unittests/test_imperative_transformer.py index 89ae3c6a39..d871d2dd85 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_transformer.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_transformer.py @@ -302,11 +302,8 @@ use_py_reader = False # if we run sync mode sync = False -if not core.is_compiled_with_cuda(): - # how many batches we use - batch_num = 50 -else: - batch_num = 5 +# how many batches we use +batch_num = 5 np.random.seed = 1 src_word_np = np.random.randint( -- GitLab