From 8b1da591fc550a40a50e574ab9247f81578d84ad Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 18 Apr 2019 16:25:04 +0200 Subject: [PATCH] assert for dataloader --- datasets/TTSDataset.py | 4 ++++ 1 file changed, 4 insertions(+) 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,