diff --git a/datasets/TTSDataset.py b/datasets/TTSDataset.py index c679a7ce..16b016f4 100644 --- a/datasets/TTSDataset.py +++ b/datasets/TTSDataset.py @@ -136,6 +136,10 @@ class MyDataset(Dataset): else: text = np.asarray( text_to_sequence(text, [self.cleaners]), dtype=np.int32) + + assert text.size > 0, self.items[idx][1] + assert wav.size > 0, self.items[idx][1] + sample = { 'text': text, 'wav': wav,