update

examples/mpnet_aishell/run.sh

@@ -8,7 +8,7 @@ sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name f
 --speech_dir "E:/programmer/asr_datasets/aishell/data_aishell/wav/train"
 
 
-sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name file_dir \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 

examples/mpnet_aishell/step_3_evaluation.py

@@ -154,9 +154,11 @@ def main():
 
         noise_audio = noise_audio.unsqueeze(dim=0)
         clean_audio = clean_audio.unsqueeze(dim=0)
-        noisy_audio = noisy_audio.unsqueeze(dim=0)
+        noisy_audio: torch.Tensor = noisy_audio.unsqueeze(dim=0)
 
         # inference
+        clean_audio = clean_audio.to(device)
+        noisy_audio = noisy_audio.to(device)
         noisy_mag, noisy_pha, noisy_com = mag_pha_stft(
             noisy_audio, config.n_fft, config.hop_size, config.win_size, config.compress_factor
         )