distributed

get_dist_info

def get_dist_info() -> Dict[str, Union[int, str]]

Get distributed training information.

Returns:

A dictionary containing rank, local_rank, world_size, and launcher.

get_rank

def get_rank() -> int

Get the rank of current process.

Returns:

The rank of current process.

get_local_rank

def get_local_rank() -> int

Get the local rank of current process.

Returns:

The local rank of current process.

get_world_size

def get_world_size() -> int

Get the world size.

Returns:

The number of processes.

get_launcher

def get_launcher() -> Optional[str]

Get the launcher type.

Returns:

The launcher type: 'torch', 'mpi', or None.