From 44f50cf4e56d5d2afbb3c99e34f6d4b2929346eb Mon Sep 17 00:00:00 2001 From: Jiabin Yang Date: Tue, 9 Apr 2019 12:16:59 +0800 Subject: [PATCH] test=release/1.4, fix test_impertative_transformer (#16708) --- .../fluid/tests/unittests/test_imperative_transformer.py | 6 +----- 1 file changed, 1 insertion(+), 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 df097360c..ef9d3ffca 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_transformer.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_transformer.py @@ -302,11 +302,7 @@ 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 +batch_num = 5 np.random.seed = 1 src_word_np = np.random.randint( -- GitLab