diff --git a/python/paddle/fluid/tests/unittests/hybrid_parallel_pp_alexnet.py b/python/paddle/fluid/tests/unittests/hybrid_parallel_pp_alexnet.py index 14d7e960f4a68cfcc7101b353eb379fee0739074..912849ffbeb71c953e9316a701e5add24ddf4440 100644 --- a/python/paddle/fluid/tests/unittests/hybrid_parallel_pp_alexnet.py +++ b/python/paddle/fluid/tests/unittests/hybrid_parallel_pp_alexnet.py @@ -113,7 +113,7 @@ class TestDistPPTraning(unittest.TestCase): print("loss: ", loss_a.numpy(), loss_b.numpy()) np.testing.assert_allclose( - loss_a.numpy(), loss_b.numpy(), rtol=1e-5) + loss_a.numpy(), loss_b.numpy(), rtol=5e-5) if __name__ == "__main__":