f-dangel / curvlinops / 12485821515

90%

push

github

[DEL] Remove `to_device` function, run deterministic tests on original device (#160)

* [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

* [ADD] Implement `FisherMCLinearOperator` as `CurvatureLinearOperator`

* [ADD] Implement Jacobians as `CurvatureLinearOperator`

* [REF] Extract error test with ``MutableMapping`` and ``batch_size_fn=None``

* [ADD] Class attribute for checking deterministic batches

* [ADD] Extract getters for `_in_shape` and `_out_shape`

* [ADD] Test deterministic batches

* [DOC] Add documentation and type annotations

* [ADD] Implement empirical Fisher as `CurvatureLinearOperator`

* [REF] Extract testing matrix multiplies in expectation

* [FIX] Examples

* [ADD] Check deterministic batches in MC-Fisher (fixes #75)

* [ADD] Implement activation Hessian as `CurvatureLinearOperator`

* [ADD] Implement KFAC as `CurvatureLinearOperator`

* [DEL] Remove original base class

* [FIX] Darglint

* [DEL] Remove unused imports

* [ADD] Declare activation Hessian as self-adjoint

* [DEL] Remove `to_device` function, run deterministic tests on original dev

* [FIX] Device issue in activation Hessian test

* [FIX] Some tests

* [FIX] KFAC tests

* [FIX] Implement KFAC inverse as Curvature linop

* [FIX] Make tests work on GPU and CPU

* Tweak tolerances for KFAC-MC tests on GPU

* [FIX] Linters

* [FIX] Test

* [DEL] Blank line

* Adjust numerical tolerances for tests

* [REF] Fix nit

* [DOC] Update changelog

---------

... (continued)

6 of 6 new or added lines in 2 files covered. (100.0%)

1384 of 1483 relevant lines covered (93.32%)

0.93 hits per line