提交 bee68872 authored 作者: Frederic's avatar Frederic

Make test more resistent to different seed value.

上级 dc5617c1
......@@ -78,12 +78,15 @@ def test_cholesky_grad():
rng = numpy.random.RandomState(utt.fetch_seed())
r = rng.randn(5, 5).astype(config.floatX)
pd = numpy.dot(r, r.T)
eps = None
if config.floatX == "float64":
eps = 2e-8
# Check the default.
yield utt.verify_grad, cholesky, [pd], 3, rng
utt.verify_grad(cholesky, [pd], 3, rng, eps=eps)
# Explicit lower-triangular.
yield utt.verify_grad, Cholesky(lower=True), [pd], 3, rng
utt.verify_grad(Cholesky(lower=True), [pd], 3, rng, eps=eps)
# Explicit upper-triangular.
yield utt.verify_grad, Cholesky(lower=False), [pd], 3, rng
utt.verify_grad(Cholesky(lower=False), [pd], 3, rng, eps=eps)
def test_cholesky_and_cholesky_grad_shape():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论