change probs tolerance to 1e-4

2690f7f0 · pangyoki · 32210b08 · 2690f7f0
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_distribution.py python/paddle/fluid/tests/unittests/test_distribution.py +2 -1

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_distribution.py
+++ b/python/paddle/fluid/tests/unittests/test_distribution.py
@@ -408,7 +408,8 @@ class NormalTest(unittest.TestCase):
            entropy, np_entropy, rtol=tolerance, atol=tolerance)
        np.testing.assert_allclose(
            log_prob, np_lp, rtol=log_tolerance, atol=log_tolerance)
-        np.testing.assert_allclose(probs, np_p, rtol=tolerance, atol=tolerance)
+        np.testing.assert_allclose(
+            probs, np_p, rtol=log_tolerance, atol=log_tolerance)
        np.testing.assert_allclose(kl, np_kl, rtol=tolerance, atol=tolerance)

    def test_normal_distribution_dygraph(self, sample_shape=7, tolerance=1e-6):