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