update

toolbox/torchaudio/losses/spectral.py

@@ -184,6 +184,11 @@ class SpectralConvergenceLoss(torch.nn.Module):
             raise AssertionError("SpectralConvergenceLoss, inf in truth_norm")
 
         batch_loss = error_norm / truth_norm
+        if torch.any(torch.isnan(batch_loss)):
+            raise AssertionError("SpectralConvergenceLoss, nan in batch_loss")
+        if torch.any(torch.isinf(batch_loss)):
+            raise AssertionError("SpectralConvergenceLoss, inf in batch_loss")
+
         if self.reduction == "mean":
             loss = torch.mean(batch_loss)
         elif self.reduction == "sum":