UserDefinedRoleMaker¶

class paddle.distributed.fleet. UserDefinedRoleMaker ( is_collective: bool = False, init_gloo: bool = False, **kwargs: Any ) [source]

UserDefinedRoleMaker is an interface for distributed configuration initialization based on obtaining distributed related information from user-defined parameters.

Examples

>>> 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"])