mirror of https://github.com/coqui-ai/TTS.git
Define metafiles in conf.json
This commit is contained in:
parent
b83d125e52
commit
f24a9e74b8
|
@ -23,13 +23,15 @@
|
||||||
"griffin_lim_iters": 60,
|
"griffin_lim_iters": 60,
|
||||||
"power": 1.5,
|
"power": 1.5,
|
||||||
|
|
||||||
"num_loader_workers": 14,
|
"num_loader_workers": 8,
|
||||||
|
|
||||||
"checkpoint": true,
|
"checkpoint": true,
|
||||||
"save_step": 750,
|
"save_step": 750,
|
||||||
"print_step": 10,
|
"print_step": 10,
|
||||||
"run_eval": false,
|
"run_eval": false,
|
||||||
"data_path": "/snakepit/shared/data/keithito/LJSpeech-1.1/",
|
"data_path": "/snakepit/shared/data/keithito/LJSpeech-1.1/",
|
||||||
|
"meta_file_train": "metadata.csv",
|
||||||
|
"meta_file_val": "metadata_val.csv",
|
||||||
"min_seq_len": 0,
|
"min_seq_len": 0,
|
||||||
"output_path": "experiments/"
|
"output_path": "experiments/"
|
||||||
}
|
}
|
||||||
|
|
4
train.py
4
train.py
|
@ -332,7 +332,7 @@ def main(args):
|
||||||
max_mel_freq = c.max_mel_freq)
|
max_mel_freq = c.max_mel_freq)
|
||||||
|
|
||||||
# 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.text_cleaner,
|
c.text_cleaner,
|
||||||
|
@ -346,7 +346,7 @@ def main(args):
|
||||||
pin_memory=True)
|
pin_memory=True)
|
||||||
|
|
||||||
if c.run_eval:
|
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'),
|
os.path.join(c.data_path, 'wavs'),
|
||||||
c.r,
|
c.r,
|
||||||
c.text_cleaner,
|
c.text_cleaner,
|
||||||
|
|
Loading…
Reference in New Issue