mmengine.dist.get_world_size

mmengine.dist.get_world_size(group=None)

Return the number of the given process group.

Note

Calling get_world_size in non-distributed environment will return 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 the number of processes of the given process group if in distributed environment, otherwise 1.

Return type:

int