diff --git a/python/paddle/fluid/tests/unittests/test_bpr_loss_op.py b/python/paddle/fluid/tests/unittests/test_bpr_loss_op.py index d137f4a6fbe4953618a0c355d224a78407482a43..80916f4a8287f576142f141eeb88ed2bb5f32cbb 100644 --- a/python/paddle/fluid/tests/unittests/test_bpr_loss_op.py +++ b/python/paddle/fluid/tests/unittests/test_bpr_loss_op.py @@ -25,7 +25,7 @@ class TestBprLossOp1(OpTest): def setUp(self): self.op_type = "bpr_loss" - batch_size = 4 + batch_size = 40 class_num = 5 X = randomize_probability(batch_size, class_num, dtype='float64') label_pos = np.random.randint(