From 9d34c540d96b2b26d3cccea5f0346727cda8ca83 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Fri, 7 Oct 2022 21:56:55 +0000 Subject: [PATCH] in memory dataset off by default --- dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataset.py b/dataset.py index 274dc3d..322bada 100644 --- a/dataset.py +++ b/dataset.py @@ -146,7 +146,7 @@ class DatasetFactory(): return tf_signature - def get_dataset(self, folder, batch_size, signature, shuffle=True, use_filled=False, resolutions=[512], in_memory=True): + def get_dataset(self, folder, batch_size, signature, shuffle=True, use_filled=False, resolutions=[512], in_memory=False): # assert (use_filled and len(resolutions) >= 1) or (not use_filled and len(resolutions) == 1) assert set(resolutions).issubset(set([32, 64, 128, 256, 512]))