search

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360 stars 43 forks source link

Compute the most negative eigenvalues with power iteration (untested) #14

Open noahgolmant opened 5 years ago

noahgolmant commented 5 years ago

This adds a flag to compute the most negative eigenvalues for looking at negative curvature.

Resolves #11