how to train the model with docker