From 58dbabd5611941335a6d68404fffdbccee56437f Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 31 Aug 2022 17:33:07 +0100 Subject: [PATCH] fix another crash --- aimodel/src/lib/dataset/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py index 59e03d9..3fac94d 100644 --- a/aimodel/src/lib/dataset/dataset.py +++ b/aimodel/src/lib/dataset/dataset.py @@ -49,7 +49,7 @@ def dataset(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_m dataset_train = make_dataset(filepaths_train, batch_size=batch_size, parallel_reads_multiplier=parallel_reads_multiplier) dataset_validate = make_dataset(filepaths_validate, batch_size=batch_size, parallel_reads_multiplier=parallel_reads_multiplier) - return dataset_train, dataset_validate, filepaths + return dataset_train, dataset_validate #, filepaths def dataset_predict(): raise NotImplementedError("Not implemented yet")