cerebras.modelzoo.common.run_utils.run#
- cerebras.modelzoo.common.run_utils.run(extra_args_parser_fn=None)[source]#
Entry point to running pytorch models including CLI argument parsing.
- Parameters
extra_args_parser_fn (Optional[Callable[[], List[argparse.ArgumentParser]]]) – An optional callable that adds any extra parser args not covered in get_parser fn.