From 9efc72db7394b7ee61f04b3c8fe2b53ef224a447 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 16 Jun 2023 18:36:18 +0100 Subject: [PATCH] dlr ds/mono: just why --- aimodel/src/lib/dataset/dataset_mono.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_mono.py b/aimodel/src/lib/dataset/dataset_mono.py index e46635c..76b9eca 100644 --- a/aimodel/src/lib/dataset/dataset_mono.py +++ b/aimodel/src/lib/dataset/dataset_mono.py @@ -175,7 +175,7 @@ def dataset_mono_predict(dirpath_input, batch_size=64, **kwargs): return make_dataset( filepaths=filepaths, metadata=read_metadata(dirpath_input), - batch_size=None, # NOTE: We IGNORE batch_size because otherwise we have issues reading the dataset which is stupid because it doesn't always happen and it's weird. + batch_size=batch_size, # WAS None shuffle=False, #even with shuffle=False we're not gonna get them all in the same order since we're reading in parallel **kwargs )