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

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
a06f4b2b · Jiabin Yang · GitHub · 55b15db5 · a06f4b2b
隐藏空白更改
内联并排

Showing with 2 addition and 5 deletion

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

未找到文件。
--- 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
-    # how many batches we use
+batch_num = 5
-    batch_num = 50
-else:
-    batch_num = 5
 np.random.seed = 1
 src_word_np = np.random.randint(