bug fix in preprocessor

pull/384/head
Eren Gölge 2021-03-16 17:04:34 +01:00
parent 01e35e06c4
commit 4c1aed4a9c
1 changed files with 1 additions and 1 deletions

View File

@ -35,7 +35,7 @@ def load_meta_data(datasets, eval_split=True):
meta_data_eval_all += meta_data_eval
meta_data_train_all += meta_data_train
# load attention masks for duration predictor training
if 'meta_file_attn_mask' in dataset:
if 'meta_file_attn_mask' in dataset and dataset['meta_file_attn_mask'] is not None:
meta_data = dict(load_attention_mask_meta_data(dataset['meta_file_attn_mask']))
for idx, ins in enumerate(meta_data_train_all):
attn_file = meta_data[ins[1]].strip()