Reset threshold in test case.

PaddlePaddle · Aug 15, 2023 · e84c04f · e84c04f
1 parent 97b6180
commit e84c04f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/collective/fleet/hybrid_parallel_mp_broadcast_obj.py b/test/collective/fleet/hybrid_parallel_mp_broadcast_obj.py
@@ -162,7 +162,7 @@ def test_mp_model_async_allreduce(self):
             loss_b = self.train_batch(batch, model_b, optimizer_b, False)
 
             np.testing.assert_allclose(
-                loss_a.numpy(), loss_b.numpy(), rtol=1e-6
+                loss_a.numpy(), loss_b.numpy(), rtol=1e-4
             )