Update data_utils.py

This commit is contained in:
Stardust·减
2023-08-21 07:31:36 +08:00
committed by GitHub
parent eca560ed82
commit 518f2e2483

View File

@@ -29,6 +29,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
self.win_length = hparams.win_length self.win_length = hparams.win_length
self.sampling_rate = hparams.sampling_rate self.sampling_rate = hparams.sampling_rate
self.spk_map = hparams.spk2id self.spk_map = hparams.spk2id
self.hparams = hparams
self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False) self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False)
if self.use_mel_spec_posterior: if self.use_mel_spec_posterior: