import argparse

parser = argparse.ArgumentParser()

parser.add_argument('--vocab_size', type=int, default=10000)
parser.add_argument('--max_len', type=int, default=50)
parser.add_argument('--word_dropout_rate', type=float, default=0.2)
parser.add_argument('--batch_size', type=int, default=100)
parser.add_argument('--encoder_embedding_dim', type=int, default=256)
parser.add_argument('--decoder_embedding_dim', type=int, default=256)
parser.add_argument('--rnn_cell', type=str, default='gru')
parser.add_argument('--rnn_size', type=int, default=256)
parser.add_argument('--encoder_layers', type=int, default=2)
parser.add_argument('--decoder_layers', type=int, default=2)
parser.add_argument('--beam_width', type=int, default=5)
parser.add_argument('--clip_norm', type=float, default=5.0)
parser.add_argument('--latent_size', type=int, default=16)
parser.add_argument('--anneal_max', type=float, default=1.0)
parser.add_argument('--anneal_bias', type=int, default=6000)
parser.add_argument('--mutinfo_weight', type=float, default=1.0)
parser.add_argument('--num_epoch', type=int, default=10)
parser.add_argument('--display_loss_step', type=int, default=10)
parser.add_argument('--display_text_step', type=int, default=100)

args = parser.parse_args()
