We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a109d5c commit 43c8f57Copy full SHA for 43c8f57
gpytorch/utils/qr.py
@@ -29,6 +29,6 @@ def stable_qr(mat):
29
Rdiag_sign = torch.sign(Rdiag)
30
# force zero diagonals to have jitter added to them.
31
Rdiag_sign[Rdiag_sign == 0] = 1.0
32
- jitter_diag = 1e-4 * Rdiag_sign * zeroish.to(Rdiag)
+ jitter_diag = 1e-6 * Rdiag_sign * zeroish.to(Rdiag)
33
R = R + torch.diag_embed(jitter_diag)
34
return Q, R
0 commit comments