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]))