From 526d716d7799c20e6bd94a7283e8ab1741d34adf Mon Sep 17 00:00:00 2001 From: "Mohamad H. Danesh" Date: Mon, 23 Mar 2020 16:18:35 -0700 Subject: [PATCH 1/2] Update train_variational_autoencoder_pytorch.py --- train_variational_autoencoder_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_variational_autoencoder_pytorch.py b/train_variational_autoencoder_pytorch.py index 5a662a9..bda329f 100644 --- 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'][::] From 898ffd0aa3bfde5fd8a4bc4c45e9630e2edb5e0e Mon Sep 17 00:00:00 2001 From: "Mohamad H. Danesh" Date: Mon, 23 Mar 2020 16:19:33 -0700 Subject: [PATCH 2/2] Update train_variational_autoencoder_tensorflow.py --- train_variational_autoencoder_tensorflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_variational_autoencoder_tensorflow.py b/train_variational_autoencoder_tensorflow.py index 922b83f..b7531f2 100644 --- 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)