Shortcuts

mmengine.dist.is_main_process

mmengine.dist.is_main_process(group=None)[source]

Whether the current rank of the given process group is equal to 0.

Parameters:

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

Returns:

Return True if the current rank of the given process group is equal to 0, otherwise False.

Return type:

bool