/Hessian-Eigen-Pytorch

A tool to analyze and plot the local curvature information of a neural-network, implemented with pytorch

Primary LanguagePython

Watchers