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,