Update data_utils.py

2023-08-21 07:31:36 +08:00
parent eca560ed82
commit 518f2e2483
1 changed files with 1 additions and 0 deletions
--- a/data_utils.py
+++ b/data_utils.py
@@ -29,6 +29,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
        self.win_length = hparams.win_length
        self.sampling_rate = hparams.sampling_rate
        self.spk_map = hparams.spk2id
+        self.hparams = hparams

        self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False)
        if self.use_mel_spec_posterior: