Merge pull request #26 from modanesh/master

Make data file name consistent
jaanli · Mar 30, 2020 · 9683413 · 9683413
2 parents 2c702b5 + 898ffd0
commit 9683413
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train_variational_autoencoder_pytorch.py b/train_variational_autoencoder_pytorch.py
@@ -154,7 +154,7 @@ def load_binary_mnist(cfg, **kwcfg):
   if not fname.exists():
     print('Downloading binary MNIST data...')
     data.download_binary_mnist(fname)
-  f = h5py.File(pathlib.os.path.join(pathlib.os.environ['DAT'], 'binarized_mnist.hdf5'), 'r')
+  f = h5py.File(pathlib.os.path.join(pathlib.os.environ['DAT'], 'binary_mnist.h5'), 'r')
   x_train = f['train'][::]
   x_val = f['valid'][::]
   x_test = f['test'][::]

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)