import argparse


def get_param():
    parser = argparse.ArgumentParser()
    parser.add_argument("--epoch", type=int, help="how many epoch will you train?")
    parser.add_argument("--batch_size", type=int, help="size of batch")
    parser.add_argument("--learning_rate", type=float, default=0.002, help="learning_rate using ADAM optimizer")
    parser.add_argument("--continue_from", type=str, default=None, help="continue training from: {your trained weight}")
    parser.add_argument("--mini_epoch_discriminator", type=int, default=4,
                        help="how many epochs does discriminator trains over a epoch?")
    parser.add_argument("--mini_epoch_generator", type=int, default=4,
                        help="how many epochs does generator trains over a epoch?")
    parser.add_argument("--AttentiveRNNBLCKs", type=int, default=3,
                        help="how many LSTM blocks in generator?")
    parser.add_argument("--AttentiveRNNResNetdepth", type=int, default=2,
                        help="how deep is each RNN blocks?")
    parser.add_argument("--save_interval", type=int, default=10,
                        help="weight save interval")
    parser.add_argument("--sample_interval", type=int, default=10,
                        help="sample image interval")
    return parser