メインコンテンツまでスキップ
バージョン: v2506

utils

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.

replace_params_to_file

def replace_params_to_file(input_filepath: str, output_filepath: str,
params: Dict[str, Any]) -> None

Replace parameters in a file with given values.

Arguments:

  • input_filepath str - Path to the input file.
  • output_filepath str - Path to the output file.
  • params Dict[str, Any] - Dictionary of parameters to replace.