From ead2a74bf076460ea3e4df5b503dcbbaaeeb0269 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 23 Dec 2021 16:12:16 -0700 Subject: [PATCH] Add debug_failures flag --- codes/data/audio/paired_voice_audio_dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/codes/data/audio/paired_voice_audio_dataset.py b/codes/data/audio/paired_voice_audio_dataset.py index dd7367cb..dbdf0f98 100644 --- a/codes/data/audio/paired_voice_audio_dataset.py +++ b/codes/data/audio/paired_voice_audio_dataset.py @@ -60,6 +60,7 @@ class TextWavLoader(torch.utils.data.Dataset): self.load_conditioning = opt_get(hparams, ['load_conditioning'], False) self.conditioning_candidates = opt_get(hparams, ['num_conditioning_candidates'], 3) self.conditioning_length = opt_get(hparams, ['conditioning_length'], 44100) + self.debug_failures = opt_get(hparams, ['debug_loading_failures'], False) self.audiopaths_and_text = [] for p, fm in zip(self.path, fetcher_mode): if fm == 'lj' or fm == 'libritts': @@ -127,7 +128,8 @@ class TextWavLoader(torch.utils.data.Dataset): tseq, wav, text, path = self.get_wav_text_pair(self.audiopaths_and_text[index]) cond = self.load_conditioning_candidates(self.audiopaths_and_text[index][0]) if self.load_conditioning else None except: - print(f"error loading {self.audiopaths_and_text[index][0]}") + if self.debug_failures: + print(f"error loading {self.audiopaths_and_text[index][0]}") return self[index+1] if wav is None or \ (self.max_wav_len is not None and wav.shape[-1] > self.max_wav_len) or \