mmengine.dist.get_world_size — mmengine 0.11.0rc0 documentation (original) (raw)

• Docs >
• mmengine.dist >
• mmengine.dist.get_world_size
• 以中文阅读

Shortcuts

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

• mmengine.dist.get_world_size
  • get_world_size()