Skip to content

Commit 16ff8ce

Browse files
authored
Lanczos README update!
1 parent ec5dc5e commit 16ff8ce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,14 @@ eigenvals, eigenvecs = compute_hessian_eigenthings(model, dataloader,
2727
loss, num_eigenthings)
2828
```
2929

30-
This also includes a more general power iteration with deflation implementation in `power_iter.py`.
30+
This also includes a more general power iteration with deflation implementation in `power_iter.py`. `lanczos.py` calls a [`scipy` hook](https://docs.scipy.org/doc/scipy-0.14.0/reference/generated/scipy.sparse.linalg.eigsh.html) to a battle-tested ARPACK implementation.
3131

3232
## Citing this work
3333
If you find this repo useful and would like to cite it in a publication, here is a BibTeX entry:
3434

3535
@misc{hessian-eigenthings,
3636
author = {Noah Golmant, Zhewei Yao, Amir Gholami, Michael Mahoney, Joseph Gonzalez},
37-
title = {{pytorch-hessian-eigentings: efficient PyTorch Hessian eigendecomposition}},
37+
title = {pytorch-hessian-eigentings: efficient PyTorch Hessian eigendecomposition},
3838
month = oct,
3939
year = 2018,
4040
version = {1.0},

0 commit comments

Comments
 (0)