def main():
data_path = os.path.join(CONFIGS['data']['dir'],
CONFIGS['data']['filename'])
data = Data(data_path, split=[0.7, 0.15, 0.15])
data.preprocess()
data.vectorize()
trainloader = DataLoader(data.train(), batch_size=128)
trainloader2 = DataLoader(data.train(), batch_size=1)
devloader = DataLoader(data.valid(), batch_size=1)
vae = VAE(data.input_dim_(), 256, 128, device).to(device)
vae.train(trainloader, trainloader2, devloader)
