add annotation file names to config

This commit is contained in:
Eren Golge 2018-04-17 10:02:45 -07:00
Родитель 1aad10c31c
Коммит c0bbdf7fe0
2 изменённых файлов: 4 добавлений и 2 удалений

Просмотреть файл

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