diff --git a/TTS/tts/datasets/TTSDataset.py b/TTS/tts/datasets/TTSDataset.py index 3d85e000..eaabb42b 100644 --- a/TTS/tts/datasets/TTSDataset.py +++ b/TTS/tts/datasets/TTSDataset.py @@ -53,6 +53,7 @@ class MyDataset(Dataset): use_noise_augment (bool): enable adding random noise to wav for augmentation. verbose (bool): print diagnostic information. """ + super(MyDataset, self).__init__() self.batch_group_size = batch_group_size self.items = meta_data self.outputs_per_step = outputs_per_step diff --git a/TTS/vocoder/datasets/wavernn_dataset.py b/TTS/vocoder/datasets/wavernn_dataset.py index 257800b0..d45932c9 100644 --- a/TTS/vocoder/datasets/wavernn_dataset.py +++ b/TTS/vocoder/datasets/wavernn_dataset.py @@ -21,6 +21,7 @@ class WaveRNNDataset(Dataset): verbose=False, ): + super(WaveRNNDataset, self).__init__() self.ap = ap self.compute_feat = not isinstance(items[0], (tuple, list)) self.item_list = items