update

examples/spectrum_dfnet_aishell/run.sh

@@ -12,7 +12,7 @@ sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name fi
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 
-sh run.sh --stage 2 --stop_stage 3 --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/spectrum_dfnet_aishell/step_3_evaluation.py

@@ -231,6 +231,8 @@ def main():
         snr_db = torch.mean(snr_db, dim=1, keepdim=True)
         # snr_db shape: [batch_size, 1, time_steps]
 
+        speech_spec_complex = speech_spec_complex.to(device)
+        mix_spec_complex = mix_spec_complex.to(device)
         mix_spec = mix_spec.to(device)
         speech_irm_target = speech_irm.to(device)
         snr_db_target = snr_db.to(device)