update

examples/spectrum_dfnet_aishell/step_3_evaluation.py

@@ -105,12 +105,12 @@ def enhance(mix_spec_complex: torch.Tensor,
     mask_speech = speech_irm_prediction
     mask_noise = 1.0 - speech_irm_prediction
 
-    # print(f"mix_spec_complex: {mix_spec_complex.shape}")
-    # print(f"mask_noise: {mask_noise.shape}")
-
     speech_spec = mix_spec_complex * mask_speech
     noise_spec = mix_spec_complex * mask_noise
 
+    print(f"speech_spec_prediction: {speech_spec_prediction.shape}")
+    print(f"noise_spec: {noise_spec.shape}")
+
     speech_wave = istft.forward(speech_spec_prediction)
     # speech_wave = istft.forward(speech_spec)
     noise_wave = istft.forward(noise_spec)