f-dangel / curvlinops / 6813760752

93%

main: 90%

push

github

[ADD] Support for KFAC with type-2 Fisher (#56)

* Add test for MSELoss type-2 KFAC

* Implement type-2 KFAC for MSELoss

* Add docstrings

* Fix black

* Fix docstring

* Add test for CrossEntropyLoss type-2 KFAC

* Implement type-2 KFAC for CrossEntropyLoss

* Fix auto-merge issue

* Fix comment

* [REF] Refactor type-2 using Hessian matrix square root

* [DEL] Remove unused imports

* [FIX] Darglint

* [FIX] Function name in docs

* [REF] Improve function name

* [REF] Rename `num_classes` into `output_dim`

---------

Co-authored-by: Felix Dangel <felix.dangel@vectorinstitute.ai>

40 of 45 new or added lines in 2 files covered. (88.89%)

791 of 854 relevant lines covered (92.62%)

0.93 hits per line