def add_training_arguments(parser):
    """Add training-specific arguments from the command line

    Parameters
    ----------
    parser : argparse.ArgumentParser
        Input parser

    Returns
    -------
    argparse.ArgumentParser
        Modified parser, with additional arguments
    """
    parser.add_argument('-t', '--nb-training-image',
                        type=int,
                        default=0,
                        help=("Number of training images"))
    parser.add_argument('-v', '--nb-validation-image',
                        type=int,
                        default=0,
                        help=("Number of validation images"))
    return parser