Define metafiles in conf.json

pull/10/head
Eren G 2018-07-25 12:34:42 +02:00
parent b83d125e52
commit f24a9e74b8
2 changed files with 5 additions and 3 deletions

View File

@ -23,13 +23,15 @@
"griffin_lim_iters": 60,
"power": 1.5,
"num_loader_workers": 14,
"num_loader_workers": 8,
"checkpoint": true,
"save_step": 750,
"print_step": 10,
"run_eval": false,
"data_path": "/snakepit/shared/data/keithito/LJSpeech-1.1/",
"meta_file_train": "metadata.csv",
"meta_file_val": "metadata_val.csv",
"min_seq_len": 0,
"output_path": "experiments/"
}

View File

@ -332,7 +332,7 @@ def main(args):
max_mel_freq = c.max_mel_freq)
# 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.text_cleaner,
@ -346,7 +346,7 @@ def main(args):
pin_memory=True)
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'),
c.r,
c.text_cleaner,