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)