From 95c2a2ae071baa504defb285663b35241b26b1fe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Mon, 21 Aug 2023 08:59:06 +0800 Subject: [PATCH] Update data_utils.py --- data_utils.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/data_utils.py b/data_utils.py index 0c81725..cdda3aa 100644 --- a/data_utils.py +++ b/data_utils.py @@ -131,14 +131,14 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset): word2ph[i] = word2ph[i] * 2 word2ph[0] += 1 bert_path = wav_path.replace(".wav", ".bert.pt") - # try: - bert = torch.load(bert_path) - assert bert.shape[-1] == len(phone) - # except: - # bert = get_bert(text, word2ph, language_str) - # torch.save(bert, bert_path) - # print(bert.shape[-1], bert_path, text, pold) - # assert bert.shape[-1] == len(phone) + try: + bert = torch.load(bert_path) + assert bert.shape[-1] == len(phone) + except: + bert = get_bert(text, word2ph, language_str) + torch.save(bert, bert_path) + print(bert.shape[-1], bert_path, text, pold) + assert bert.shape[-1] == len(phone) assert bert.shape[-1] == len(phone), ( bert.shape, len(phone), sum(word2ph), p1, p2, t1, t2, pold, pold2, word2ph, text, w2pho)