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

添加spec生成失败捕获

Browse Source
This commit is contained in:
Stardust·减
2023-09-04 21:14:43 +08:00
committed by GitHub
parent 5914ed3efc
commit 1ef312f5b1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data_utils.py
View File

@@ -89,9 +89,9 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
spec_filename = filename.replace(".wav", ".spec.pt") spec_filename = filename.replace(".wav", ".spec.pt")
if self.use_mel_spec_posterior: if self.use_mel_spec_posterior:
spec_filename = spec_filename.replace(".spec.pt", ".mel.pt") spec_filename = spec_filename.replace(".spec.pt", ".mel.pt")
if os.path.exists(spec_filename): try:
spec = torch.load(spec_filename) spec = torch.load(spec_filename)
else: except:
if self.use_mel_spec_posterior: if self.use_mel_spec_posterior:
# if os.path.exists(filename.replace(".wav", ".spec.pt")): # if os.path.exists(filename.replace(".wav", ".spec.pt")):
# # spec, n_fft, num_mels, sampling_rate, fmin, fmax # # spec, n_fft, num_mels, sampling_rate, fmin, fmax