issues
search
noahgolmant
/
pytorch-hessian-eigenthings
Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360
stars
43
forks
source link
Add power iteration accuracy tests with minor stability fixes for power iteration
#26
Closed
noahgolmant
closed
5 years ago
noahgolmant
commented
5 years ago
This adds tests for
Power iteration on random PSD Wishart matrices
Power iteration using the full dataset for simple linear models
Power iteration using a fixed mini-batch
Stochastic power iteration using a new mini-batch for each hessian-vector product evaluation
Plots can be found in #23.
This adds tests for
Plots can be found in #23.