From 39d0099990d095cf12b92e904e34945a5cf17969 Mon Sep 17 00:00:00 2001 From: Fieldhunter <1677532160Yuan@gmail.com> Date: Sun, 9 Feb 2020 16:57:48 +0800 Subject: [PATCH] Update 5.2_best_practice.py add define server --- code/5_control_flow_analysis/5.2_best_practice.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/code/5_control_flow_analysis/5.2_best_practice.py b/code/5_control_flow_analysis/5.2_best_practice.py index f4ace92..5000e03 100644 --- a/code/5_control_flow_analysis/5.2_best_practice.py +++ b/code/5_control_flow_analysis/5.2_best_practice.py @@ -69,6 +69,7 @@ def main(unused_argv): "ps": ps_spec, "worker": worker_spec}) + server = tf.train.Server(cluster, job_name=FLAGS.job_name, task_index=FLAGS.task_index) # 如果是ps,直接启动服务,并开始监听worker发起的请求 if FLAGS.job_name == "ps": server.join() @@ -238,4 +239,4 @@ def main(unused_argv): if __name__ == "__main__": - tf.app.run() \ No newline at end of file + tf.app.run()