update

toolbox/torchaudio/losses/spectral.py

@@ -183,6 +183,10 @@ class SpectralConvergenceLoss(torch.nn.Module):
             loss = torch.sum(batch_loss)
         else:
             raise AssertionError
+
+        if torch.any(torch.isnan(loss)) or torch.any(torch.isinf(loss)):
+            raise AssertionError("SpectralConvergenceLoss, nan or inf in loss")
+
         return loss
 
 
@@ -209,7 +213,13 @@ class LogSTFTMagnitudeLoss(torch.nn.Module):
         :param clean_magnitude: Tensor, shape: [batch_size, time_steps, freq_bins]
         :return:
         """
-        return F.l1_loss(torch.log(denoise_magnitude + self.eps), torch.log(clean_magnitude + self.eps))
+
+        loss = F.l1_loss(torch.log(denoise_magnitude + self.eps), torch.log(clean_magnitude + self.eps))
+
+        if torch.any(torch.isnan(loss)) or torch.any(torch.isinf(loss)):
+            raise AssertionError("SpectralConvergenceLoss, nan or inf in loss")
+
+        return loss
 
 
 class STFTLoss(torch.nn.Module):