Bug fixes and a more comprehensive training script
Automated training with boosting stages