Add utility to track the Hessian throughout training (WIP)

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

MIT License

360 stars 43 forks source link

Open noahgolmant opened 5 years ago

noahgolmant commented 5 years ago

This PR should:

Provide tracking class to update estimates with incremental power iteration (done, untested)
Add example of tracking throughout training (done, untested)
Add README information on tracking

Addresses #7