Merge pull request #16658 from JiabinYang/fix/transformer_random_failed

test=develop, fix transformer in dygraph

Merge pull request #16658 from JiabinYang/fix/transformer_random_failed
test=develop, fix transformer in dygraph
38382f8e · lujun · GitHub · 61735850 · 4c18b98f · 38382f8e
显示空白变更内容
内联并排

Showing with 5 addition and 2 deletion

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

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_imperative_transformer.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_transformer.py
@@ -302,8 +302,11 @@ use_py_reader = False
 # if we run sync mode
 sync = False
-# how many batches we use
+if not core.is_compiled_with_cuda():
-batch_num = 50
+    # how many batches we use
+    batch_num = 50
+else:
+    batch_num = 5
 np.random.seed = 1
 src_word_np = np.random.randint(