mmengine.dist.get_world_size¶

mmengine.dist.get_world_size(group=None)[source]¶

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