mirror of https://github.com/coqui-ai/TTS.git
add annotation file names to config
This commit is contained in:
parent
b27d2eebd5
commit
14c508a61a
|
@ -21,6 +21,8 @@
|
||||||
"power": 1.2,
|
"power": 1.2,
|
||||||
|
|
||||||
"dataset": "TWEB",
|
"dataset": "TWEB",
|
||||||
|
"meta_file_train": "transcript_train.txt",
|
||||||
|
"meta_file_val": "transcript_val.txt"
|
||||||
"data_path": "/run/shm/erogol/BibleSpeech/",
|
"data_path": "/run/shm/erogol/BibleSpeech/",
|
||||||
"min_seq_len": 0,
|
"min_seq_len": 0,
|
||||||
"num_loader_workers": 8,
|
"num_loader_workers": 8,
|
||||||
|
|
4
train.py
4
train.py
|
@ -296,7 +296,7 @@ def main(args):
|
||||||
Dataset = getattr(mod, c.dataset+"Dataset")
|
Dataset = getattr(mod, c.dataset+"Dataset")
|
||||||
|
|
||||||
# Setup the 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'),
|
os.path.join(c.data_path, 'wavs'),
|
||||||
c.r,
|
c.r,
|
||||||
c.sample_rate,
|
c.sample_rate,
|
||||||
|
@ -317,7 +317,7 @@ def main(args):
|
||||||
drop_last=False, num_workers=c.num_loader_workers,
|
drop_last=False, num_workers=c.num_loader_workers,
|
||||||
pin_memory=True)
|
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'),
|
os.path.join(c.data_path, 'wavs'),
|
||||||
c.r,
|
c.r,
|
||||||
c.sample_rate,
|
c.sample_rate,
|
||||||
|
|
Loading…
Reference in New Issue