From 7541d2ecaaa7d159c23cd05b9097c8b2bb7c2769 Mon Sep 17 00:00:00 2001 From: erogol Date: Wed, 25 Nov 2020 14:50:09 +0100 Subject: [PATCH] return eval split optional --- TTS/tts/datasets/preprocess.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/TTS/tts/datasets/preprocess.py b/TTS/tts/datasets/preprocess.py index 31d4b2b5..01a56adb 100644 --- a/TTS/tts/datasets/preprocess.py +++ b/TTS/tts/datasets/preprocess.py @@ -20,13 +20,13 @@ def load_meta_data(datasets, eval_split=True): preprocessor = get_preprocessor_by_name(name) meta_data_train = preprocessor(root_path, meta_file_train) print(f" | > Found {len(meta_data_train)} files in {Path(root_path).resolve()}") - if meta_file_val is None: - meta_data_eval, meta_data_train = split_dataset(meta_data_train) - else: - meta_data_eval = preprocessor(root_path, meta_file_val) - meta_data_train_all += meta_data_train - if meta_data_eval_all is not None: + if eval_split: + if meta_file_val is None: + meta_data_eval, meta_data_train = split_dataset(meta_data_train) + else: + meta_data_eval = preprocessor(root_path, meta_file_val) meta_data_eval_all += meta_data_eval + meta_data_train_all += meta_data_train return meta_data_train_all, meta_data_eval_all