From db5af92afdc9702ddf282147318298a9152f3e8e Mon Sep 17 00:00:00 2001 From: phlrain Date: Fri, 4 Mar 2022 05:16:37 +0000 Subject: [PATCH] set bs = 5; test=develop --- .../unittests/test_imperative_transformer_sorted_gradient.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_imperative_transformer_sorted_gradient.py b/python/paddle/fluid/tests/unittests/test_imperative_transformer_sorted_gradient.py index a4fabad60c8..010c8aeccac 100644 --- 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( -- GitLab