def parse_args(parser):
    """
    Parse commandline arguments.
    """

    parser.add_argument('--bench-class',  type=str, choices=['train', 'perf-infer', 'perf-train'], required=True, help='Choose test class')

    return parser