coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

assert for dataloader

pull/10/head
Eren Golge 2019-04-18 16:25:04 +02:00
parent 4f65b2907e
commit 8b1da591fc
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
datasets/TTSDataset.py
View File

@ -136,6 +136,10 @@ class MyDataset(Dataset):
else: else:
text = np.asarray( text = np.asarray(
text_to_sequence(text, [self.cleaners]), dtype=np.int32) 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 = { sample = {
'text': text, 'text': text,
'wav': wav, 'wav': wav,