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