mmengine.dist.get_dist_info

mmengine.dist.get_dist_info(group=None)

Get distributed information of the given process group.

备注

Calling get_dist_info in non-distributed environment will return (0, 1).

参数:

group (ProcessGroup, optional) – The process group to work on. If None, the default process group will be used. Defaults to None.

返回:

Return a tuple containing the rank and world_size.

返回类型:

tuple[int, int]