def get_base_argument_parser()

in quant/common/parser.py [0:0]

• 26 lines of code
• 1 McCabe index (conditional complexity)

def get_base_argument_parser(description: str) -> ArgumentParser:
    """
    Get a base argument parser for driver scripts.

    Args:
        description: A string describing the driver script.

    Returns:
        Parser object to extend.
    """
    parser = ArgumentParser(description)
    parser.add_argument('--config', type=str, help='Path to a yaml config file.')
    parser.add_argument(
        '--experiment-name', type=str, default=None, help='Name of the experiment.'
    )
    parser.add_argument(
        '--ngpus', type=int, default=None, help='Number of GPUs. Use 0 for CPU.'
    )
    parser.add_argument(
        '--skip-training',
        default=False,
        action='store_true',
        help='Skip training and only run evaluation. Checkpoint must be passed in as well.',
    )
    parser.add_argument(
        '--restore-experiment',
        type=str,
        help='Path to experiments directory to restore checkpoint from.',
    )
    parser.add_argument(
        '--init-from-checkpoint',
        type=str,
        help='Path to model file to initialize model parameters.',
    )
    return parser