HoneyTian commited on
Commit
8125774
·
1 Parent(s): eedc28d
toolbox/torchaudio/losses/perceptual.py CHANGED
@@ -85,7 +85,12 @@ class PesqLoss(nn.Module):
85
  )
86
 
87
  def forward(self, denoise: torch.Tensor, clean: torch.Tensor):
88
-
 
 
 
 
 
89
  batch_loss = self.loss_fn.forward(clean, denoise)
90
 
91
  mask = ~(torch.isnan(batch_loss) | torch.isinf(batch_loss))
 
85
  )
86
 
87
  def forward(self, denoise: torch.Tensor, clean: torch.Tensor):
88
+ max_val = torch.max(
89
+ torch.amax(denoise.abs(), dim=1, keepdim=True),
90
+ torch.amax(clean.abs(), dim=1, keepdim=True),
91
+ )
92
+ if max_val == 0:
93
+ raise AssertionError
94
  batch_loss = self.loss_fn.forward(clean, denoise)
95
 
96
  mask = ~(torch.isnan(batch_loss) | torch.isinf(batch_loss))