← all repositories

tomgoldstein/loss-landscape

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

3.2k stars Python ML FrameworksLLMOps · Eval
loss-landscape
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.