Module training_viz 

Training visualization: loss landscapes, gradient flow, weight distributions, dashboards.

Structs

ActivationMapViz

Render intermediate activations as colored grids.

EpochStats

Statistics for a single training epoch.

GradientFlowViz

Gradient magnitude per layer, rendered as a horizontal bar chart.

LossLandscape

A 2D grid of loss values sampled along two parameter directions.

TrainingDashboard

Composite training dashboard combining multiple visualizations.

TrainingLog

Log of training progress over multiple epochs.

WeightDistViz

Render weight histograms per layer.

WeightStats

Functions

render_loss_landscape

Render a loss landscape as colored points: (position, loss_value, color).