update

toolbox/torchaudio/losses/perceptual.py

@@ -85,10 +85,7 @@ class PesqLoss(nn.Module):
         )
 
     def forward(self, denoise: torch.Tensor, clean: torch.Tensor):
-        max_val = torch.max(
-            torch.amax(denoise.abs(), dim=1, keepdim=True),
-            torch.amax(clean.abs(), dim=1, keepdim=True),
-        )
+        max_val = torch.max(clean.abs())
         if max_val == 0:
             raise AssertionError
         batch_loss = self.loss_fn.forward(clean, denoise)