def add_generate_args(parser):
    """Add generation-related arguments.

    Args:
    * parser: argument parser

    Returns:
    * parser: argument parser with generation-related arguments inserted
    """

    generate_arg = parser.add_argument_group('Generate')
    generate_arg.add_argument('--no_norm', dest="auto_norm",
                              action="store_false", default=True,
                              help='Wether to norm the input labels')
    generate_arg.add_argument('--encode_times',
                              type=int, default=10, metavar='N',
                              help='Encode smile N times.')
    generate_arg.add_argument('--decode_times',
                              type=int, default=5, metavar='M',
                              help='Decode each latent vector M times. There will be N*M candidates.')

    return parser