f-dangel / curvlinops

89%

main: 89%

LAST BUILD ON BRANCH kfac-state-dict

branch: kfac-state-dict

CHANGE BRANCH

x

Reset

Sync Branches

• kfac-state-dict
• activation-hessian
• activation-hessian-linop
• adjoint
• adjoint-jacobians
• adjoint-submatrix
• alt-ekfac
• always-full-tests
• arbitrary-inputs
• bcewithlogitsloss
• bug-dtype-check-deterministic
• bug-kfac-device-change
• ci-update
• composition-rules
• copilot/add-fishertype-kfactype-enums
• curvature-linop
• deprecate-pkg-resources
• deprecate-python38
• development
• diagonal
• doc-eigenvalues
• dtype-device-properties
• ef-linop
• ekfac
• empirical-fisher
• eva
• example-benchmark
• example-fisher-mc
• example-trace
• expand-reduce
• faster-ef
• faster-fisher-mc
• fisher-averaging
• fisher-linop
• fisher-mc
• fix-coveralls-badge
• fix-device
• fix-device-errors
• fix-enum
• fix-inverse-kfac
• fix-kfac-scale
• fix-markdown-syntax
• fix-mc-tests
• fix-mcfisher-scale
• fix-python-badge
• fix-scale
• fix-typo
• flake8-to-ruff
• foof
• frobenius-norm
• functional-estimators
• functional-ggnmp
• functorch-efmp
• functorch-fmp
• functorch-ggnmp
• functorch-hmp
• functorch-jacobians
• ggn-linop
• hessian-blocks
• hessian-linop
• high-dim-outputs
• hutchinson
• hutchpp
• improve-benchmark-results
• improve-frobenius
• improve-spectral-density
• improve-tests
• improve-visual-tour
• issue-138-improve-links
• jacobian
• jacobian-linop
• jacobian-transposed
• joint-weight-and-bias
• kfac
• kfac-arbitrary-params-order
• kfac-ce-loss
• kfac-conv2d
• kfac-damping
• kfac-ef
• kfac-ema
• kfac-inv
• kfac-linop
• kfac-torch-only
• kfac-trace
• kfac-type-2
• kfac-with-inplace-activations
• lanczos-cache
• load-state-dict-model-func
• logo
• lsmr-inverse
• main
• matmat-list-shapes
• matvec-to-matmat
• move-util-linops
• neumann-inverse
• neumann-inverse-example
• no-cpu
• only-bias-or-weight
• overwrite-model-warning
• papyan20traces-density
• parallelize-grad-output
• parallelize-sample-grad-output
• parallelize-tests
• partial-boundary-specification
• power-iteration-comparison
• pydocstyle-coverage
• pytorch-cg
• pytorch-inverse
• pytorch-linop
• pytorch-spectral-density
• reduce-scipy
• ref-internal-mapping
• remove-kfac-side-effect
• remove-loss-average
• remove-to_device
• remover-iterator-conversion
• retry-kfac-inv
• rla-pytorch
• runame-patch-1
• source-button
• specify-num-data
• submatrices
• submatrix-example
• submatrix-pytorch
• test-consecutive-matmats
• test-regression
• update-benchmark-a6000
• update-black
• update-changelog
• update-ci
• update-github-actions
• use-built-in-functorch
• use-pyproject-toml
• use-tuples-in-functionals
• v3-release-notes
• visual-tour
• weighted-ce-kfac
• xdiag
• xtrace

Build # 9214642589

Build Type

Pull #114

github

Committed by runame

Commit Message

Merge branch 'main' into kfac-state-dict

Pull Request Pull Request #114: [ADD] `state_dict` functionality to `KFACLinearOperator` and `KFACInverseLinearOperator`

Run Details

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