From 8307ad12e626b6c88e675a4d8faaa60f24d2fff3 Mon Sep 17 00:00:00 2001 From: Natanael Date: Mon, 21 Aug 2023 17:34:53 -0300 Subject: [PATCH] no cache --- mergernet/data/dataset.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mergernet/data/dataset.py b/mergernet/data/dataset.py index d906fce1..339da472 100644 --- 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')