assert for dataloader

2026-09-09 11:16:00 +08:00 · 2019-04-18 16:25:04 +02:00
parent 4f65b2907e
commit 8b1da591fc
1 changed files with 4 additions and 0 deletions
@@ -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,