def train(self, model_builder, job_args):
"""Runs the training process to train a model.
Arguments:
model_builder: the ModelBuilder to use to build graphs during training.
job_args: the arguments for the training job.
Returns:
The trained Model. The resulting value is only relevant for master nodes.
"""
job = Job(model_builder, job_args.output, self._config)
job.configure_logging()
server = self._config.create_server()
if server and self._config.param_server:
return self._run_ps(server)
return self._run_training(server, job)
