Update data_utils.py

2023-08-21 07:25:24 +08:00
parent 627d755042
commit a0547bb383
1 changed files with 4 additions and 0 deletions
--- a/data_utils.py
+++ b/data_utils.py
@@ -30,6 +30,10 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
        self.sampling_rate = hparams.sampling_rate
        self.spk_map = hparams.spk2id

+        self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False)
+        if self.use_mel_spec_posterior:
+            self.n_mel_channels = getattr(hparams, "n_mel_channels", 80)
+
        self.cleaned_text = getattr(hparams, "cleaned_text", False)

        self.add_blank = hparams.add_blank