From 52b699db80b71416e532e7d62a555ba1fa4ac763 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sat, 2 Sep 2023 00:54:29 +0800 Subject: [PATCH] Update data_utils.py --- data_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/data_utils.py b/data_utils.py index 3a369e8..f6ebfe8 100644 --- a/data_utils.py +++ b/data_utils.py @@ -135,10 +135,10 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset): 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) + 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)