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

44f50cf4 · Jiabin Yang · GitHub · a0af374f · 44f50cf4
隐藏空白更改
内联并排

Showing with 1 addition and 5 deletion

python/paddle/fluid/tests/unittests/test_imperative_transformer.py ...ddle/fluid/tests/unittests/test_imperative_transformer.py +1 -5

未找到文件。
--- 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(