π Pico Training Metrics Dashboard
A beautiful, interactive web dashboard for visualizing training progress across all your Pico model runs.
β¨ Features
- π Training Loss Visualization: Track loss curves over time for all runs
- π― Learning Rate Schedules: Monitor LR progression and warmup patterns
- π Paloma Evaluation: View perplexity metrics during training
- π Combined View: See all metrics together for easy comparison
- π¨ Interactive Charts: Built with Chart.js for smooth interactions
- π± Responsive Design: Works on desktop and mobile devices
- βοΈ Run Comparison: Compare different model configurations side-by-side
οΏ½οΏ½ Quick Start
Generate Data: First, run the data generation script to parse your training logs:
python scripts/generate_data.pyView the Dashboard: Open
index.htmlin your web browserSelect Runs: Use the dropdown to view specific runs or all runs together
Toggle Metrics: Check/uncheck boxes to show/hide different metric types
Explore Charts: Hover over data points for detailed information
π Files
index.html- Main dashboard interfacestyle.css- Modern, responsive stylingcode.js- Interactive chart functionalitydata.json- Training metrics data (auto-generated from logs)
π§ Data Source
The dashboard automatically extracts training metrics from:
- Training loss at each step
- Learning rate progression
- Paloma evaluation results
- Model configuration parameters
π Updating Data
To refresh the dashboard with new training data:
- Run new training sessions - logs will be saved to
runs/*/logs/ - Generate updated data.json:
python scripts/generate_data.py - Refresh the dashboard - new runs will appear automatically
π¨ Chart Types
- Training Loss: Line charts showing loss reduction over time
- Learning Rate: Logarithmic scale for LR schedule visualization
- Evaluation: Paloma perplexity metrics during training
- Combined: All metrics on one chart for easy comparison
π‘ Usage Tips
- Compare Runs: Select "All Runs" to see how different configurations perform
- Zoom In: Use the chart zoom features to focus on specific training phases
- Export: Right-click charts to save as images
- Mobile: Dashboard is fully responsive for mobile devices
π― Key Metrics Tracked
- Training Loss: Primary performance indicator
- Learning Rate: Schedule adherence and warmup progress
- Paloma Perplexity: Model evaluation quality
- Inf/NaN Counts: Training stability monitoring
- Model Config: Architecture and hyperparameter details
π Design Features
- Modern UI: Clean, professional interface
- Color Coding: Distinct colors for each model run
- Responsive Layout: Adapts to different screen sizes
- Interactive Elements: Hover effects and smooth animations
- Professional Typography: Easy-to-read fonts and spacing
π Documentation
For more details on generating the data.json file, see:
scripts/README.md- Complete script documentationscripts/generate_data.py- The data generation script
Built with β€οΈ for the Pico Language Model training community