Parallelize the training process