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.