mmengine.dist.get_dist_info

mmengine.dist.get_dist_info(group=None)

Get distributed information of the given process group.

Note

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

Parameters:

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

Returns:

Return a tuple containing the rank and world_size.

Return type:

tuple[int, int]