From 44d8c7d53d6f60fa59c8ee614bcca44d920b21ea Mon Sep 17 00:00:00 2001 From: Caleb Weinreb Date: Sat, 29 Jul 2023 08:59:53 -0400 Subject: [PATCH] reduce diag boost --- dynamax/utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dynamax/utils/utils.py b/dynamax/utils/utils.py index b78d3421..e75d6f01 100644 --- a/dynamax/utils/utils.py +++ b/dynamax/utils/utils.py @@ -199,7 +199,7 @@ def find_permutation( return perm -def psd_solve(A, b, diagonal_boost=1e-6): +def psd_solve(A, b, diagonal_boost=1e-9): """A wrapper for coordinating the linalg solvers used in the library for psd matrices.""" A = symmetrize(A) + diagonal_boost * jnp.eye(A.shape[-1]) L, lower = cho_factor(A, lower=True)