From f036e79098628ba1c7c1c16a5d9411c81dd1a5b9 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 15 Sep 2022 17:09:26 +0100 Subject: [PATCH] fixup --- aimodel/src/lib/dataset/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py index afa8d27..8241e34 100644 --- a/aimodel/src/lib/dataset/dataset.py +++ b/aimodel/src/lib/dataset/dataset.py @@ -84,7 +84,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_m return dataset_train, dataset_validate #, filepaths -def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5, pretrain=False): +def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5, prefetch=False): filepaths = get_filepaths(dirpath_input) filepaths_count = len(filepaths) for i in range(len(filepaths)): @@ -96,7 +96,7 @@ def dataset_predict(dirpath_input, batch_size=64, parallel_reads_multiplier=1.5, batch_size=batch_size, parallel_reads_multiplier=parallel_reads_multiplier, dummy_label=False, - pretrain=pretrain + prefetch=prefetch ), filepaths[0:filepaths_count], filepaths_count if __name__ == "__main__":