Warn user if nan in GL

2021-09-10 08:26:05 +00:00 · 2021-09-10 08:26:05 +00:00 · 742f9c54da
parent ed4b1d8514
commit 742f9c54da
1 changed files with 3 additions and 0 deletions
--- a/TTS/utils/audio.py
+++ b/TTS/utils/audio.py
@ -608,6 +608,9 @@ class AudioProcessor(object):
        angles = np.exp(2j * np.pi * np.random.rand(*S.shape))
        S_complex = np.abs(S).astype(np.complex)
        y = self._istft(S_complex * angles)
+        if not np.isfinite(y).all():
+            print(" [!] Waveform is not finite everywhere. Skipping the GL.")
+            return np.array([0])
        for _ in range(self.griffin_lim_iters):
            angles = np.exp(1j * np.angle(self._stft(y)))
            y = self._istft(S_complex * angles)