Skip to content

Commit 526d716

Browse files
authored
Update train_variational_autoencoder_pytorch.py
1 parent 2c702b5 commit 526d716

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_variational_autoencoder_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def load_binary_mnist(cfg, **kwcfg):
154154
if not fname.exists():
155155
print('Downloading binary MNIST data...')
156156
data.download_binary_mnist(fname)
157-
f = h5py.File(pathlib.os.path.join(pathlib.os.environ['DAT'], 'binarized_mnist.hdf5'), 'r')
157+
f = h5py.File(pathlib.os.path.join(pathlib.os.environ['DAT'], 'binary_mnist.h5'), 'r')
158158
x_train = f['train'][::]
159159
x_val = f['valid'][::]
160160
x_test = f['test'][::]

0 commit comments

Comments
 (0)