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 7e18913a03b79e38b662b946e540ab58af0487ac..2af6461aedb28889e57cf2b0101030579afae33a 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 = 3 + batch_size = 4 class_num = 5 X = randomize_probability(batch_size, class_num, dtype='float64') label_pos = np.random.randint(