cerebras.modelzoo.common.utils.run.cli_parser.add_general_arguments#

cerebras.modelzoo.common.utils.run.cli_parser.add_general_arguments(parser, modes=['train', 'eval', 'train_and_eval', 'eval_all'])[source]#

Injects general parser arguments.

Parameters

  • parser (argparse.ArgumentParser) – Parser into which the arguments are being added.

  • modes (List[str]) – Optional list of valid modes to be passed under the –mode argument of the parser. We default to choices [“train”, “eval”, “train_and_eval”, “eval_all”]. If an empty list if provided, then –mode isn’t added as a parser arg.

previous

cerebras.modelzoo.common.utils.run.cli_parser.add_csx_arguments

next

cerebras.modelzoo.common.utils.run.cli_parser.add_gpu_arguments