From e9964e7fad6cfad3a3b87d5d8013eeab7a56929e Mon Sep 17 00:00:00 2001 From: Eren G Date: Wed, 25 Jul 2018 12:34:42 +0200 Subject: [PATCH] Define metafiles in conf.json --- config.json | 4 +++- train.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index 3a371e60..e2b0cb4d 100644 --- a/config.json +++ b/config.json @@ -23,13 +23,15 @@ "griffin_lim_iters": 60, "power": 1.5, - "num_loader_workers": 14, + "num_loader_workers": 8, "checkpoint": true, "save_step": 750, "print_step": 10, "run_eval": false, "data_path": "/snakepit/shared/data/keithito/LJSpeech-1.1/", + "meta_file_train": "metadata.csv", + "meta_file_val": "metadata_val.csv", "min_seq_len": 0, "output_path": "experiments/" } diff --git a/train.py b/train.py index 9f62f003..697a233d 100644 --- a/train.py +++ b/train.py @@ -332,7 +332,7 @@ def main(args): max_mel_freq = c.max_mel_freq) # Setup the dataset - train_dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata_train.csv'), + train_dataset = LJSpeechDataset(os.path.join(c.data_path, c.meta_file_train), os.path.join(c.data_path, 'wavs'), c.r, c.text_cleaner, @@ -346,7 +346,7 @@ def main(args): pin_memory=True) if c.run_eval: - val_dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata_val.csv'), + val_dataset = LJSpeechDataset(os.path.join(c.data_path, c.meta_file_val), os.path.join(c.data_path, 'wavs'), c.r, c.text_cleaner,