Skip to content

Commit 44f50cf

Browse files
authored
test=release/1.4, fix test_impertative_transformer (#16708)
1 parent a0af374 commit 44f50cf

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

python/paddle/fluid/tests/unittests/test_imperative_transformer.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -302,11 +302,7 @@ def make_all_inputs(input_fields):
302302
# if we run sync mode
303303
sync = False
304304

305-
if not core.is_compiled_with_cuda():
306-
# how many batches we use
307-
batch_num = 50
308-
else:
309-
batch_num = 5
305+
batch_num = 5
310306

311307
np.random.seed = 1
312308
src_word_np = np.random.randint(

0 commit comments

Comments
 (0)