Skip to content

Commit

Permalink
Change default damping value
Browse files Browse the repository at this point in the history
  • Loading branch information
runame committed Feb 6, 2024
1 parent 7a4a36e commit 73ee54b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion curvlinops/inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,14 +212,15 @@ class KFACInverseLinearOperator(_InverseLinearOperator):
def __init__(
self,
A: KFACLinearOperator,
damping: Optional[Tuple[float, float]] = None,
damping: Tuple[float, float] = (0., 0.),
cache: bool = True,
):
"""Store the linear operator whose inverse should be represented.
Args:
A: ``KFACLinearOperator`` whose inverse is formed.
damping: Damping values for all input and gradient covariances.
Default: ``(0., 0.)``.
cache: Whether to cache the inverses of the Kronecker factors.
Default: ``True``.
Expand Down

0 comments on commit 73ee54b

Please sign in to comment.