tomgoldstein/loss-landscape
PyTorch-based tool for visualizing loss surfaces of neural networks near optimal parameters.

Velocity · 7d
+1.1
★ / day
Trend
→steady
star history
This repository provides code for visualizing the loss landscape of neural networks. Given a pre-trained model and its parameters, it calculates and plots loss surfaces along random directions, enabling analysis of network training dynamics. The tool supports distributed computation across multiple GPUs and nodes, storing results in HDF5 format. The work accompanied a NIPS 2018 paper on loss landscape visualization.