What problem does it solve?
Long-running machine learning training runs can fail silently due to issues like NaN loss, loss divergence, or metric degradation, often wasting hours of expensive GPU compute before the problem is discovered. This Skill automates periodic checks of training metrics to catch these issues early, preventing wasted resources and reducing debugging time.
Core Features & Use Cases
- Automated WandB Metric Monitoring: Periodically checks training loss, evaluation metrics, gradient norms, and learning rate schedules for anomalies without manual intervention.
- Intelligent Judgment & Escalation: Uses a clear decision framework to flag critical issues immediately, and escalates ambiguous cases to a reasoning model for nuanced assessment.
- Integrated Health Checks: Works alongside process watchdog tools to cover both runtime stability (session alive, GPU active) and training quality, providing full visibility into run health.
- Use Case: An ML engineer running a multi-day vertebrae segmentation model training can use this Skill to automatically monitor the run, stop it early if loss diverges, and avoid wasting days of GPU time on a broken experiment.
Quick Start
Use the training-check skill to run an automated health check on your active WandB training run <entity>/<project>/<run_id> and flag any critical issues like NaN loss or metric degradation.