update

examples/mpnet_aishell/step_3_evaluation.py

@@ -85,14 +85,14 @@ def save_audios(noise_audio: torch.Tensor,
     output_dir.mkdir(parents=True, exist_ok=True)
 
     filename = output_dir / "noise_audio.wav"
-    torchaudio.save(filename, noise_audio, sample_rate)
+    torchaudio.save(filename, noise_audio.detach().cpu(), sample_rate)
     filename = output_dir / "clean_audio.wav"
-    torchaudio.save(filename, clean_audio, sample_rate)
+    torchaudio.save(filename, clean_audio.detach().cpu(), sample_rate)
     filename = output_dir / "noisy_audio.wav"
-    torchaudio.save(filename, noisy_audio, sample_rate)
+    torchaudio.save(filename, noisy_audio.detach().cpu(), sample_rate)
 
     filename = output_dir / "enhanced_audio.wav"
-    torchaudio.save(filename, enhanced_audio, sample_rate)
+    torchaudio.save(filename, enhanced_audio.detach().cpu(), sample_rate)
 
     return output_dir.as_posix()