in memory dataset off by default

2022-10-07 21:56:55 +00:00
parent 5dd0dd3248
commit 9d34c540d9
1 changed files with 1 additions and 1 deletions
@@ -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]))