提交 0094e5fa 编写于 作者: P pangyoki

change tolerance of Normal kl method

上级 2690f7f0
......@@ -410,7 +410,8 @@ class NormalTest(unittest.TestCase):
log_prob, np_lp, rtol=log_tolerance, atol=log_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)
np.testing.assert_allclose(
kl, np_kl, rtol=log_tolerance, atol=log_tolerance)
def test_normal_distribution_dygraph(self, sample_shape=7, tolerance=1e-6):
paddle.disable_static(self.place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册