set bs = 5; test=develop

db5af92a · phlrain · b8d52a82 · db5af92a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_imperative_transformer_sorted_gradient.py .../unittests/test_imperative_transformer_sorted_gradient.py +1 -1

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_imperative_transformer_sorted_gradient.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_transformer_sorted_gradient.py
@@ -315,7 +315,7 @@ use_py_reader = False
 sync = False

 # how many batches we use
-batch_num = 2
+batch_num = 5

 np.random.seed(90)
 src_word_np = np.arange(1, TrainTaskConfig.batch_size * seq_len + 1).reshape(