tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update data_utils.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-21 08:59:06 +08:00
committed by GitHub
parent 58adc54f07
commit 95c2a2ae07
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
data_utils.py
View File

@@ -131,14 +131,14 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
word2ph[i] = word2ph[i] * 2 word2ph[i] = word2ph[i] * 2
word2ph[0] += 1 word2ph[0] += 1
bert_path = wav_path.replace(".wav", ".bert.pt") bert_path = wav_path.replace(".wav", ".bert.pt")
# try: try:
bert = torch.load(bert_path) bert = torch.load(bert_path)
assert bert.shape[-1] == len(phone) assert bert.shape[-1] == len(phone)
# except: except:
# bert = get_bert(text, word2ph, language_str) bert = get_bert(text, word2ph, language_str)
# torch.save(bert, bert_path) torch.save(bert, bert_path)
# print(bert.shape[-1], bert_path, text, pold) print(bert.shape[-1], bert_path, text, pold)
# assert bert.shape[-1] == len(phone) assert bert.shape[-1] == len(phone)
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) bert.shape, len(phone), sum(word2ph), p1, p2, t1, t2, pold, pold2, word2ph, text, w2pho)