f-dangel / curvlinops / 9214967831

90%

push

github

[ADD] `state_dict` functionality to `KFACLinearOperator` and `KFACInverseLinearOperator` (#114)

* Add tests for state dict functionality

* Add state dict functionality to (inverse) KFAC linear operator

* Fix tests

* Address review comments on tests

* Test torch.save/load as well and fix order of equivalence checks

* Check if covariance and mapping keys match when loading state dict

* Use compare_state_dicts everywhere

62 of 65 new or added lines in 2 files covered. (95.38%)

1324 of 1492 relevant lines covered (88.74%)

0.89 hits per line