wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update data loader tests

Browse Source
This commit is contained in:
Eren Gölge
2022-02-25 11:05:54 +01:00
parent 93957d58a1
commit 90cc45dd4e
1 changed files with 0 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/datasets/dataset.py
-17
View File
@@ -69,9 +69,6 @@ class TTSDataset(Dataset):
samples (list): List of dataset samples.
tokenizer (TTSTokenizer): tokenizer to convert text to sequence IDs. If None init internally else
use the given. Defaults to None.
tokenizer (TTSTokenizer): tokenizer to convert text to sequence IDs. If None init internally else
use the given. Defaults to None.
@@ -205,20 +202,6 @@ class TTSDataset(Dataset):
token_ids = self.tokenizer.text_to_ids(text)
return np.array(token_ids, dtype=np.int32)
@staticmethod
def _parse_sample(item):
language_name = None
attn_file = None
if len(item) == 5:
text, wav_file, speaker_name, language_name, attn_file = item
elif len(item) == 4:
text, wav_file, speaker_name, language_name = item
elif len(item) == 3:
text, wav_file, speaker_name = item
else:
token_ids = self.tokenizer.text_to_ids(text)
return token_ids
def load_data(self, idx):
item = self.samples[idx]