Simplify and modularize the training pipeline #19

Closed mazabou closed 4 days ago