diff --git a/TTS/vocoder/datasets/gan_dataset.py b/TTS/vocoder/datasets/gan_dataset.py index c101791a..41744fd9 100644 --- a/TTS/vocoder/datasets/gan_dataset.py +++ b/TTS/vocoder/datasets/gan_dataset.py @@ -69,7 +69,8 @@ class GANDataset(Dataset): cache acoustic features """ # set the seed differently for each worker - random.seed(torch.utils.data.get_worker_info().seed) + if torch.utils.data.get_worker_info(): + random.seed(torch.utils.data.get_worker_info().seed) if self.return_segments: item1 = self.load_item(idx)