update

toolbox/torchaudio/losses/spectral.py

@@ -175,7 +175,14 @@ class SpectralConvergenceLoss(torch.nn.Module):
         :return:
         """
         error_norm = torch.norm(denoise_magnitude - clean_magnitude, p="fro", dim=(-1, -2))
+        if torch.any(torch.isnan(error_norm)) or torch.any(torch.isinf(error_norm)):
+            raise AssertionError("SpectralConvergenceLoss, nan or inf in error_norm")
         truth_norm = torch.norm(clean_magnitude, p="fro", dim=(-1, -2))
+        if torch.any(torch.isnan(truth_norm)):
+            raise AssertionError("SpectralConvergenceLoss, nan in truth_norm")
+        if torch.any(torch.isinf(truth_norm)):
+            raise AssertionError("SpectralConvergenceLoss, inf in truth_norm")
+
         batch_loss = error_norm / truth_norm
         if self.reduction == "mean":
             loss = torch.mean(batch_loss)