Training models in a continuous integration pipeline