mel-spec transform infinity values bug fix

ivanvovk · Nov 1, 2020 · bc75d3a · bc75d3a
1 parent 6be2f4c
commit bc75d3a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/data.py b/data.py
@@ -72,5 +72,7 @@ def __init__(self, **kwargs):
         self.torchaudio_backend = MelSpectrogram(**kwargs)
 
     def forward(self, x):
-        outputs = self.torchaudio_backend(x)
-        return outputs[..., :-1].log10()
+        outputs = self.torchaudio_backend(x).log10()
+        mask = torch.isinf(outputs)
+        outputs[mask] = 0
+        return outputs[..., :-1]