diff --git a/params.py b/params.py index 490d401bf..4e9ce4cba 100644 --- a/params.py +++ b/params.py @@ -1,6 +1,6 @@ -batch_size = 64 # Batch size for training. +batch_size = 96 # Batch size for training. epochs = 1 # Number of epochs to train for. -latent_dim = 256 # Latent dimensionality of the encoding space. +latent_dim = 512 # Latent dimensionality of the encoding space. num_samples = 10000 # Number of samples to train on. # Path to the data txt file on disk. data_path = "fra.txt"