update

examples/conv_tasnet/step_2_train_model.py

@@ -250,6 +250,9 @@ def main():
             denoise_audios = model.forward(noisy_audios)
             denoise_audios = torch.squeeze(denoise_audios, dim=1)
 
+            if torch.any(torch.isnan(denoise_audios)) or torch.any(torch.isinf(denoise_audios)):
+                raise AssertionError("nan or inf in denoise_audios")
+
             ae_loss = ae_loss_fn.forward(denoise_audios, clean_audios)
             neg_si_snr_loss = neg_si_snr_loss_fn.forward(denoise_audios, clean_audios)
             neg_stoi_loss = neg_stoi_loss_fn.forward(denoise_audios, clean_audios)