stop writing intermediate training data (causes huge system bloat)
added simple stage feature