update

examples/clean_unet_aishell/step_2_train_model.py

@@ -235,6 +235,11 @@ def main():
 
             enhanced_audios = model.forward(noisy_audios)
             enhanced_audios = torch.squeeze(enhanced_audios, dim=1)
+
+            if torch.any(torch.isnan(clean_audios)) or torch.any(torch.isinf(clean_audios)):
+                raise AssertionError("nan or inf in clean_audios")
+            if torch.any(torch.isnan(noisy_audios)) or torch.any(torch.isinf(noisy_audios)):
+                raise AssertionError("nan or inf in noisy_audios")
             if torch.any(torch.isnan(enhanced_audios)) or torch.any(torch.isinf(enhanced_audios)):
                 raise AssertionError("nan or inf in enhanced_audios")