no cache

nmcardoso · Aug 21, 2023 · 8307ad1 · 8307ad1
1 parent b3452f3
commit 8307ad1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mergernet/data/dataset.py b/mergernet/data/dataset.py
@@ -375,7 +375,6 @@ def get_preds_dataset(
     self,
     prepare: bool = True,
     batch_size: int = 64,
-    buffer_size: int = 1000
   ) -> tf.data.Dataset:
     iaunames = self.get_X()
     paths = self.get_images_paths(iaunames)
@@ -399,8 +398,8 @@ def get_preds_dataset(
       L.info(f'Image min value: {example_X.min()}')
       L.info(f'Image datatype: {str(example_X.dtype)}')
 
-      ds = ds.cache()
-      L.info('Apply: cache')
+      # ds = ds.cache()
+      # L.info('Apply: cache')
 
       ds = ds.batch(batch_size)
       L.info('Apply: batch')