mmengine.dist.get_world_size

mmengine.dist.get_world_size(group=None)

Return the number of the given process group.

备注

Calling get_world_size in non-distributed environment will return 1.

参数

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

返回

Return the number of processes of the given process group if in distributed environment, otherwise 1.

返回类型

int