Skip to content

Commit 898ffd0

Browse files
authored
Update train_variational_autoencoder_tensorflow.py
1 parent 526d716 commit 898ffd0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_variational_autoencoder_tensorflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def train():
135135
sess = tfc.InteractiveSession()
136136
sess.run(init_op)
137137

138-
mnist_data = tfds.load(name='binarized_mnist', split='train', shuffle_files=False)
138+
mnist_data = tfds.load(name='binary_mnist', split='train', shuffle_files=False)
139139
dataset = mnist_data.repeat().shuffle(buffer_size=1024).batch(FLAGS.batch_size)
140140

141141
print('Saving TensorBoard summaries and images to: %s' % FLAGS.logdir)

0 commit comments

Comments
 (0)