添加spec生成失败捕获

2023-09-04 21:14:43 +08:00
parent 5914ed3efc
commit 1ef312f5b1
1 changed files with 2 additions and 2 deletions
--- a/data_utils.py
+++ b/data_utils.py
@@ -89,9 +89,9 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
        spec_filename = filename.replace(".wav", ".spec.pt")
        if self.use_mel_spec_posterior:
            spec_filename = spec_filename.replace(".spec.pt", ".mel.pt")
-        if os.path.exists(spec_filename):
+        try:
            spec = torch.load(spec_filename)
-        else:
+        except:
            if self.use_mel_spec_posterior:
                # if os.path.exists(filename.replace(".wav", ".spec.pt")):
                #     # spec, n_fft, num_mels, sampling_rate, fmin, fmax