End to end model training

shape