UserDefinedRoleMaker¶

class paddle.distributed.fleet. UserDefinedRoleMaker [源代码] ¶

UserDefinedRoleMaker 是基于从用户自定义的参数中获取分布式相关信息进行分布式配置初始化的接口它会自动根据用户的自定义配置进行分布式训练环境初始化，目前 UserDefinedRoleMaker 支持 ParameterServer 分布式训练及 Collective 分布式训练两种模式的初始化。

代码示例¶

          >>> import paddle.distributed.fleet as fleet
>>> from paddle.distributed.fleet.base.role_maker import Role

>>> fleet.UserDefinedRoleMaker(
...     current_id=0,
...     role=Role.SERVER,
...     worker_num=2,
...     server_endpoints=["127.0.0.1:36011", "127.0.0.1:36012"])

         

方法¶

to_string()¶

将当前环境变量以字符串的形式输出

返回

string

代码示例

           import paddle.distributed.fleet as fleet
from paddle.distributed.fleet.base.role_maker import Role
role = fleet.UserDefinedRoleMaker(
    current_id=0,
    role=Role.SERVER,
    worker_num=2,
    server_endpoints=["127.0.0.1:36011", "127.0.0.1:36012"])
role.to_string()

          