Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

argument for returning no eval split

This commit is contained in:
erogol 2020-11-12 12:52:27 +01:00
parent a7aefd5c50
commit 9b0f441945
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
TTS/tts/datasets/preprocess.py
View File

@ -9,9 +9,9 @@ from tqdm import tqdm
from TTS.tts.utils.generic_utils import split_dataset from TTS.tts.utils.generic_utils import split_dataset
def load_meta_data(datasets): def load_meta_data(datasets, eval_split=True):
meta_data_train_all = [] meta_data_train_all = []
meta_data_eval_all = [] meta_data_eval_all = [] if eval_split else None
for dataset in datasets: for dataset in datasets:
name = dataset['name'] name = dataset['name']
root_path = dataset['path'] root_path = dataset['path']
@ -25,7 +25,8 @@ def load_meta_data(datasets):
else: else:
meta_data_eval = preprocessor(root_path, meta_file_val) meta_data_eval = preprocessor(root_path, meta_file_val)
meta_data_train_all += meta_data_train meta_data_train_all += meta_data_train
meta_data_eval_all += meta_data_eval if meta_data_eval_all is not None:
meta_data_eval_all += meta_data_eval
return meta_data_train_all, meta_data_eval_all return meta_data_train_all, meta_data_eval_all