f-dangel / curvlinops / 11944028535

90%

push

github

[ADD] Change GGN to purely PyTorch (#146)

* [ADD] Minimal linear operator interface for PyTorch

* [FIX] Linters

* [ADD] Replicate current base class but inherit from `PyTorchLinearOperator`

* [DEL] Unused import

* [ADD] Implement Hessian as `CurvatureLinearOperator`

* [REF] Combine full and block-diagonal matrix multiply tests

* [ADD] Automatically implement `adjoint` for self-adjoint ops

* [ADD] Implement GGN as `CurvatureLinearOperator`

* [FIX] flake8

* [FIX] Increase tolerance to make test pass

* [DEL] Unused import

* [FIX] Try using more iterations in Neumann inverse test

* [FIX] Convert `bfloat16` to `float32` before calling `.numpy`

* [FIX] Typo in docstring

* [DOC] Update changelog

19 of 20 new or added lines in 3 files covered. (95.0%)

1571 of 1710 relevant lines covered (91.87%)

0.92 hits per line