update

toolbox/torchaudio/losses/snr.py

@@ -71,7 +71,7 @@ class NegativeSISNRLoss(nn.Module):
 
         e_noise = denoise - s_target
 
-        batch_si_snr = 10 * torch.log10(torch.norm(s_target, p=2, dim=-1) ** 2 / (torch.norm(e_noise, p=2, dim=-1) ** 2 + self.eps))
+        batch_si_snr = 10 * torch.log10(torch.norm(s_target, p=2, dim=-1) ** 2 / (torch.norm(e_noise, p=2, dim=-1) ** 2 + self.eps) + self.eps)
         # si_snr shape: [batch_size,]
 
         if self.reduction == "mean":
@@ -87,7 +87,8 @@ def main():
     batch_size = 2
     signal_length = 16000
     estimated_signal = torch.randn(batch_size, signal_length)
-    target_signal = torch.randn(batch_size, signal_length)
+    # target_signal = torch.randn(batch_size, signal_length)
+    target_signal = torch.zeros(batch_size, signal_length)
 
     si_snr_loss = NegativeSISNRLoss()