From c0bbdf7fe0f30fd68f752c38b12a206172f17f7a Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 17 Apr 2018 10:02:45 -0700 Subject: [PATCH] add annotation file names to config --- config.json | 2 ++ train.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/config.json b/config.json index d7c31c12..a005e080 100644 --- a/config.json +++ b/config.json @@ -21,6 +21,8 @@ "power": 1.2, "dataset": "TWEB", + "meta_file_train": "transcript_train.txt", + "meta_file_val": "transcript_val.txt" "data_path": "/run/shm/erogol/BibleSpeech/", "min_seq_len": 0, "num_loader_workers": 8, diff --git a/train.py b/train.py index 2a8726d6..3786c0ee 100644 --- a/train.py +++ b/train.py @@ -296,7 +296,7 @@ def main(args): Dataset = getattr(mod, c.dataset+"Dataset") # 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.sample_rate, @@ -317,7 +317,7 @@ def main(args): drop_last=False, num_workers=c.num_loader_workers, pin_memory=True) - 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.sample_rate,