chore: friendly error report

This commit is contained in:
magic-akari 2023-07-14 13:15:27 +08:00 committed by YuriHead
parent 22a881223c
commit 7336590c62
1 changed files with 3 additions and 2 deletions

View File

@ -48,8 +48,9 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
filename = filename.replace("\\", "/")
audio, sampling_rate = load_wav_to_torch(filename)
if sampling_rate != self.sampling_rate:
raise ValueError("{}: {} SR doesn't match target {} SR".format(
filename, sampling_rate, self.sampling_rate))
raise ValueError(
"Sample Rate not match. Expect {} but got {} from {}".format(
self.sampling_rate, sampling_rate, filename))
audio_norm = audio / self.max_wav_value
audio_norm = audio_norm.unsqueeze(0)
spec_filename = filename.replace(".wav", ".spec.pt")