From e1ad16a213d545e65d58332f5482433bc11878eb Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 18:58:45 +0000 Subject: [PATCH] debug A --- aimodel/src/lib/dataset/dataset_encoderonly.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_encoderonly.py b/aimodel/src/lib/dataset/dataset_encoderonly.py index 16946bf..a65c0e8 100644 --- a/aimodel/src/lib/dataset/dataset_encoderonly.py +++ b/aimodel/src/lib/dataset/dataset_encoderonly.py @@ -103,7 +103,7 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain return tf.function(parse_item_inner) -def make_dataset(filepaths, compression_type="GZIP", parallel_reads_multiplier=3, shuffle_buffer_size=2**16, batch_size=64, prefetch=True, shuffle=True, filepath_heightmap=None, **kwargs): +def make_dataset(filepaths, compression_type="GZIP", parallel_reads_multiplier=3, shuffle_buffer_size=2**15, batch_size=64, prefetch=True, shuffle=True, filepath_heightmap=None, **kwargs): if "NO_PREFETCH" in os.environ: logger.info("disabling data prefetching.")