Spaces:
Running
Running
update
Browse files
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))
|