mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
add annotation file names to config
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user