diff --git a/TTS/speaker_encoder/dataset.py b/TTS/speaker_encoder/dataset.py index 6110ac4d..748f5136 100644 --- a/TTS/speaker_encoder/dataset.py +++ b/TTS/speaker_encoder/dataset.py @@ -17,6 +17,7 @@ class MyDataset(Dataset): seq_len (int): voice segment length in seconds. verbose (bool): print diagnostic information. """ + super().__init__() self.items = meta_data self.sample_rate = ap.sample_rate self.voice_len = voice_len diff --git a/TTS/vocoder/datasets/wavegrad_dataset.py b/TTS/vocoder/datasets/wavegrad_dataset.py index 22f2af32..6cd5862a 100644 --- a/TTS/vocoder/datasets/wavegrad_dataset.py +++ b/TTS/vocoder/datasets/wavegrad_dataset.py @@ -26,6 +26,7 @@ class WaveGradDataset(Dataset): use_cache=False, verbose=False): + super().__init__() self.ap = ap self.item_list = items self.seq_len = seq_len if return_segments else None