Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set different seed in gan_dataset when it is multi-workers

This commit is contained in:
Eren Gölge 2021-04-08 11:13:28 +02:00
parent 6ee211c137
commit aee24b0704
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TTS/vocoder/datasets/gan_dataset.py
View File

@ -69,6 +69,7 @@ class GANDataset(Dataset):
cache acoustic features """ cache acoustic features """
# set the seed differently for each worker # set the seed differently for each worker
if torch.utils.data.get_worker_info():
random.seed(torch.utils.data.get_worker_info().seed) random.seed(torch.utils.data.get_worker_info().seed)
if self.return_segments: if self.return_segments: