← 🧠 Machine Learning

🧠 Hyperparameter Lab

Epoch: 0
Loss:
Gradient norm:
Best loss:
FPS:
Drag — rotate · Scroll — zoom

🧠 Neural Network Hyperparameter Optimization Implementation

A glowing marker trains in real time across a 3D loss landscape carved from overlapping basins — watch how learning rate, momentum and optimizer choice decide whether it settles into the deep global minimum or gets stuck in a shallow local one.

🔬 What It Demonstrates

The marker's height is the current loss; its purple trail is the optimizer's trajectory. Vanilla SGD, momentum SGD and a simplified Adam step are all computed live from the same underlying gradient each epoch.

🎮 How to Use

Pick an optimizer, then tune learning rate, momentum and batch-size noise. Watch epoch, loss, gradient norm and best loss update live, and restart training from a fresh random point to compare settings.

💡 Did You Know?

Adam adapts its effective step size per-parameter using running estimates of the gradient's mean and variance — which is why it often needs a much larger nominal learning rate than plain SGD to move at a comparable pace.