From aee24b070423cb36065e55d933da771221371ab6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 8 Apr 2021 11:13:28 +0200 Subject: [PATCH] set different seed in gan_dataset when it is multi-workers --- TTS/vocoder/datasets/gan_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)