mindspore.set_ps_context
- mindspore.set_ps_context(**kwargs)[源代码]
设置参数服务器训练模式的上下文。
说明
参数服务器训练模式只在图模式下支持。 需要给参数服务器训练模式设置其他的环境变量。这些环境变量如下所示:
MS_SERVER_NUM:表示参数服务器数量。
MS_WORKER_NUM:表示工作进程数量。
MS_SCHED_HOST:表示调度器IP地址。
MS_SCHED_PORT:表示调度器开启的监听端口。
MS_ROLE:表示进程角色,角色列表如下:
MS_SCHED:表示调度器。
MS_WORKER:表示工作进程。
MS_PSERVER/MS_SERVER:表示参数服务器。
- 参数:
enable_ps (bool) - 表示是否启用参数服务器训练模式。只有在enable_ps设置为True后,环境变量才会生效。默认值:False。
config_file_path (string) - 配置文件路径,用于容灾恢复等, 目前参数服务器训练模式仅支持Server容灾。默认值: ‘’。
scheduler_manage_port (int) - 调度器HTTP端口,对外开放用于接收和处理用户扩容/缩容等请求。默认值:11202。
enable_ssl (bool) - 设置是否打开SSL认证。默认值:False。
client_password (str) - 用于解密客户端证书密钥的密码。默认值: ‘’。
server_password (str) - 用于解密服务端证书密钥的密码。默认值: ‘’。
- 异常:
ValueError - 输入key不是参数服务器训练模式上下文中的属性。
样例:
>>> import mindspore as ms >>> ms.set_ps_context(enable_ps=True, enable_ssl=True, client_password='123456', server_password='123456')