PyHessian

PyHessian is a PyTorch library for Hessian based analysis of neural network models. The library enables computing the following metrics:

  • Top Hessian eigenvalues
  • The trace of the Hessian matrix
  • The full Hessian Eigenvalues Spectral Density (ESD)

Code is available at https://github.com/amirgholami/PyHessian