/loss-landscape

Code for visualizing the loss landscape of neural nets

Primary LanguagePythonMIT LicenseMIT

Given a network architecture and its pre-trained parameters, this tool calculates and visualizes the loss surface along random direction(s) near the optimal parameters.

See this notebook