update

examples/conv_tasnet/run.sh

@@ -3,7 +3,7 @@
 : <<'END'
 
 
-sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name conv-tasnet-dns3-20250319 \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name conv-tasnet-dns3-20250319 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise/dns3-noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech" \
 --max_epochs 400

toolbox/torchaudio/losses/perceptual.py

@@ -88,6 +88,11 @@ class PesqLoss(nn.Module):
 
         batch_loss = self.loss_fn.forward(clean, denoise)
 
+        mask = ~(torch.isnan(batch_loss) | torch.isinf(batch_loss))
+        batch_loss = batch_loss[mask]
+        if len(batch_loss) == 0:
+            raise AssertionError
+
         if self.reduction == "mean":
             loss = torch.mean(batch_loss)
         elif self.reduction == "sum":