diff --git a/python/paddle/fluid/tests/unittests/test_imperative_transformer.py b/python/paddle/fluid/tests/unittests/test_imperative_transformer.py index 89ae3c6a39d6277f590c8f2e02f7b0ae62a1cd4a..d871d2dd8501d76c7706febb0722c512db2f3ada 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(