Define metafiles in conf.json

This commit is contained in:
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, "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/"
} }

View File

@ -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,