def main():
"""
Descrition : main module to run code
"""
parser = argparse.ArgumentParser()
parser.add_argument('--batch_size', type=int, default=24)
parser.add_argument('--epochs', type=int, default=150)
parser.add_argument('--N', type=int, default=10)
parser.add_argument('--K', type=int, default=5)
parser.add_argument('--iterations', type=int, default=1000)
parser.add_argument('--input_dims', type=int, default=64)
parser.add_argument('--download', type=bool, default=False)
parser.add_argument('--GPU_COUNT', type=int, default=1)
parser.add_argument('--logdir', type=str, default='./log')
parser.add_argument('--modeldir', type=str, default='./models')
config = parser.parse_args()
# create output dir
try:
os.makedirs(config.logdir)
os.makedirs(config.modeldir)
except OSError:
pass
trainer = Train(config)
trainer.train()
if config.generation:
trainer.generation()
def main():
parser = argparse.ArgumentParser()
parser = argparse.ArgumentParser()
parser.add_argument('--batch_size', type=int, default=64)
parser.add_argument('--epoches', type=int, default=10)
parser.add_argument('--mu', type=int, default=128)
parser.add_argument('--n_residue', type=int, default=24)
parser.add_argument('--n_skip', type=int, default=128)
parser.add_argument('--dilation_depth', type=int, default=10)
parser.add_argument('--n_repeat', type=int, default=2)
parser.add_argument('--seq_size', type=int, default=20000)
parser.add_argument('--use_gpu', type=bool, default=False)
parser.add_argument('--generation', type=bool, default=True)
config = parser.parse_args()
trainer = Train(config)
trainer.train()
if (config.generation):
trainer.generation()