update

toolbox/torchaudio/models/clean_unet/loss.py

@@ -49,6 +49,8 @@ class LogSTFTMagnitudeLoss(torch.nn.Module):
         :param y_mag: Tensor, Magnitude spectrogram of groundtruth signal (B, #frames, #freq_bins).
         :return: Tensor, Log STFT magnitude loss value.
         """
+        y_mag = torch.clamp(y_mag, min=1e-8)
+        x_mag = torch.clamp(x_mag, min=1e-8)
         return F.l1_loss(torch.log(y_mag), torch.log(x_mag))
 
 
@@ -141,7 +143,6 @@ class MultiResolutionSTFTLoss(torch.nn.Module):
         mag_loss = 0.0
         for f in self.stft_losses:
             sc_l, mag_l = f(x, y)
-            print(f"sc_l: {sc_l}, mag_l: {mag_l}")
             sc_loss += sc_l
             mag_loss += mag_l