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