cerebras.modelzoo.common.run_utils.main#

cerebras.modelzoo.common.run_utils.main(params, script=None, extra_args_parser_fn=None)[source]#

Runs a full end-to-end CS/non-CS workflow for a PyTorch model.

Parameters

  • params (Dict[str, Any]) – The parsed YAML config dictionary.

  • script (Optional[str]) – The script to run in subprocesses for distributed GPU runs.

  • extra_args_parser_fn (Optional[Callable[[], List[argparse.ArgumentParser]]]) – An optional callable that adds any extra parser args not covered in get_parser fn.

previous

cerebras.modelzoo.common.run_utils

next

cerebras.modelzoo.common.run_utils.run