test=release/1.4, fix test_impertative_transformer (#16708)

6 years ago · 44f50cf4e5
parent a0af374f01
commit 44f50cf4e5
1 changed files with 1 additions and 5 deletions
--- 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(