What problem does it solve?
training-check prevents wasted GPU compute by detecting early signs of training failure or poor quality in WandB runs while training is still in progress.
Core Features & Use Cases
- Periodic WandB metric monitoring: checks loss trends, eval metric movement, learning-rate behavior, and signals like NaN/Inf to identify divergence and stalled learning.
- Decisioning with escalation only when ambiguous: directly stops/continues for clearly bad or good signals, and escalates to an MCP Codex judgment for uncertain cases.
- Resilient fallback when WandB is unreachable: if WandB cannot be queried, it falls back to reading the training log remotely via SSH.
- Use case: run a long training job and automatically stop it when loss diverges, metrics degrade vs baseline, or numerical issues appear, instead of discovering the failure after training ends.
Quick Start
Use training-check for the WandB run <entity>/<project>/<run_id> and let it periodically evaluate training quality until it decides to STOP, CONTINUE, or WAIT.