Update data_utils.py

This commit is contained in:
Stardust·减
2023-09-06 10:11:59 +08:00
committed by GitHub
parent 5a4e71b6ce
commit 0afb102d48

View File

@@ -80,7 +80,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
spec, wav = self.get_audio(audiopath)
sid = torch.LongTensor([int(self.spk_map[sid])])
return (phones, spec, wav, sid, tone, language, bert)
return (phones, spec, wav, sid, tone, language, bertja_bert)
def get_audio(self, filename):
audio, sampling_rate = load_wav_to_torch(filename)