FIX: Distribution

interTwin-eu · Aug 8, 2023 · 5434545 · 5434545
1 parent 0db2df2
commit 5434545
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/use-cases/zebra2horse/dataloader.py b/use-cases/zebra2horse/dataloader.py
@@ -44,25 +44,25 @@ def preproc_test_fn(img, label):
         # Apply the preprocessing operations to the training data
         train_horses = (
             train_horses.map(preproc_train_fn, num_parallel_calls=tf.data.AUTOTUNE)
-            .cache()
             .shuffle(self.buffer_size)
+            .cache()
         )
         train_zebras = (
             train_zebras.map(preproc_train_fn, num_parallel_calls=tf.data.AUTOTUNE)
-            .cache()
             .shuffle(self.buffer_size)
+            .cache()
         )
 
         # Apply the preprocessing operations to the test data
         test_horses = (
             test_horses.map(preproc_test_fn, num_parallel_calls=tf.data.AUTOTUNE)
-            .cache()
             .shuffle(self.buffer_size)
+            .cache()
         )
         test_zebras = (
             test_zebras.map(preproc_test_fn, num_parallel_calls=tf.data.AUTOTUNE)
-            .cache()
             .shuffle(self.buffer_size)
+            .cache()
         )
 
         return tf.data.Dataset.zip((train_horses, train_zebras)), tf.data.Dataset.zip((test_horses, test_zebras))