diff --git a/mindspore/context.py b/mindspore/context.py index 99020f30715..0575fa282fc 100644 --- a/mindspore/context.py +++ b/mindspore/context.py @@ -845,12 +845,19 @@ def set_ps_context(**kwargs): These environment variables are listed below: MS_SERVER_NUM: Server number + MS_WORKER_NUM: Worker number + MS_SCHED_HOST: Scheduler IP address + MS_SCHED_PORT: Scheduler port + MS_ROLE: The role of this process: + MS_SCHED: represents the scheduler, + MS_WORKER: represents the worker, + MS_PSERVER: represents the Server Args: