From 5ba3bc552ffe6d38f97e26fa181d423a37a67b45 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 11:37:05 +0100 Subject: [PATCH] Removed caching of dataset Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 504bf12..13c7ed4 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -87,7 +87,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i # get dataset dataset = tf.data.Dataset.from_tensor_slices((mnist_train_x, mnist_train_y)) - dataset = dataset.shuffle(mnist_train_x.shape[0]).batch(batch_size, drop_remainder=True).map(normalize).cache() + dataset = dataset.shuffle(mnist_train_x.shape[0]).batch(batch_size, drop_remainder=True).map(normalize) # get models encoder = Encoder(zsize)