Update train_variational_autoencoder_tensorflow.py

jaanli · Mar 23, 2020 · 898ffd0 · 898ffd0
1 parent 526d716
commit 898ffd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_variational_autoencoder_tensorflow.py b/train_variational_autoencoder_tensorflow.py
@@ -135,7 +135,7 @@ def train():
   sess = tfc.InteractiveSession()
   sess.run(init_op)
 
-  mnist_data = tfds.load(name='binarized_mnist', split='train', shuffle_files=False)
+  mnist_data = tfds.load(name='binary_mnist', split='train', shuffle_files=False)
   dataset = mnist_data.repeat().shuffle(buffer_size=1024).batch(FLAGS.batch_size)
 
   print('Saving TensorBoard summaries and images to: %s' % FLAGS.logdir)