finish building the training model