未验证 提交 1a1aeff6 编写于 作者: Z zmxdream 提交者: GitHub

[heterps]fix launch_utils.py. test=develop (#37752)

* fix launch_utils.py. test=develop

* fix launch_utils.py. test=develop
上级 caff6668
...@@ -1180,18 +1180,14 @@ class ParameterServerLauncher(object): ...@@ -1180,18 +1180,14 @@ class ParameterServerLauncher(object):
_, self.current_node_ip = get_host_name_ip() _, self.current_node_ip = get_host_name_ip()
else: else:
self.current_node_ip = pod_ip self.current_node_ip = pod_ip
if not self.distribute_mode == DistributeMode.PS_HETER: assert self.current_node_ip in self.node_ips, "Can't find your local ip {%s} in args.servers and args.workers ips: {%s}" \
assert self.current_node_ip in self.node_ips, "Can't find your local ip {%s} in args.servers and args.workers ips: {%s}" \ % (self.current_node_ip, self.node_ips)
% (self.current_node_ip, self.node_ips) self.node_rank = self.node_ips.index(self.current_node_ip)
if self.current_node_ip in self.node_ips: logger.debug(
self.node_rank = self.node_ips.index(self.current_node_ip) "parsed from args: node_ips:{} current_node_ip:{} node_rank:{}".
logger.debug( format(self.node_ips, self.current_node_ip, self.node_rank))
"parsed from args: node_ips:{} current_node_ip:{} node_rank:{}".
format(self.node_ips, self.current_node_ip, self.node_rank))
def start_ps(self): def start_ps(self):
if not self.current_node_ip in self.node_ips:
return
cluster = Cluster(hdfs=None) cluster = Cluster(hdfs=None)
server_rank = 0 server_rank = 0
worker_rank = 0 worker_rank = 0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册