noahgolmant / pytorch-hessian-eigenthings

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

add shortcut to evaluate hessian over full dataset #28

Closed noahgolmant closed 5 years ago