def add_hyperband_arguments(parser):
    """
    Add arguments to a parser for sub-command "stitch"
    :param parser: argeparse object
    :return:
    """
    parser.add_argument('--train_image_dir', type=str, required=True, help=
        'Path to directory containing labeled images for training.')
    parser.add_argument('--test_image_dir', type=str, required=True, help=
        'Path to directory containing labeled images for testing the models.')
    parser.add_argument('--gpu_mode', action='store_true', help=
        'If true then cuda is on.')
    parser.add_argument('--output_dir', type=str, required=False, default=
        './debug_output', help='Output file name.')
    parser.add_argument('--batch_size', type=int, required=False, default=
        100, help='Batch size for training, default is 100.')
    parser.add_argument('--max_epochs', type=int, required=False, default=
        10, help='Epoch size for training iteration.')
    parser.add_argument('--num_workers', type=int, required=False, default=
        40, help='Epoch size for training iteration.')
    return parser
