未验证 提交 ead89b11 编写于 作者: Webbley's avatar Webbley 提交者: GitHub

bugfix in fleetrun when launching multiple machines training manually (#37274)

上级 0b250a79
...@@ -410,7 +410,7 @@ def get_ports(num, offset): ...@@ -410,7 +410,7 @@ def get_ports(num, offset):
if ports is not None: if ports is not None:
ports = list(ports) ports = list(ports)
else: else:
start_port = os.environ.get('FLAGS_START_PORT') start_port = int(os.environ.get('FLAGS_START_PORT'))
ports = range(start_port + offset, start_port + offset + num, 1) ports = range(start_port + offset, start_port + offset + num, 1)
return ports return ports
...@@ -1131,12 +1131,12 @@ class ParameterServerLauncher(object): ...@@ -1131,12 +1131,12 @@ class ParameterServerLauncher(object):
# get http_port # get http_port
if args.http_port: if args.http_port:
self.http_port = args.http_port http_port = [args.http_port]
else: else:
http_port = get_ports( http_port = get_ports(
1, self.server_num + self.worker_num + self.heter_worker_num) 1, self.server_num + self.worker_num + self.heter_worker_num)
http_ip = self.server_endpoints.split(",")[0].split(":")[0] http_ip = self.server_endpoints.split(",")[0].split(":")[0]
self.http_port = http_ip + ":" + str(http_port[0]) self.http_port = http_ip + ":" + str(http_port[0])
# check local or user define # check local or user define
self.server_endpoints_ips = [ self.server_endpoints_ips = [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册