mmengine.dist.get_rank

mmengine.dist.get_rank(group=None)

Return the rank of the given process group.

Rank is a unique identifier assigned to each process within a distributed process group. They are always consecutive integers ranging from 0 to world_size.

注解

Calling get_rank in non-distributed environment will return 0.

参数

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

返回

Return the rank of the process group if in distributed environment, otherwise 0.

返回类型

int