HoneyTian commited on
Commit
f33a053
·
1 Parent(s): 6e4b372
examples/spectrum_dfnet_aishell/step_3_evaluation.py CHANGED
@@ -105,6 +105,9 @@ def enhance(mix_spec_complex: torch.Tensor,
105
  mask_speech = speech_irm_prediction
106
  mask_noise = 1.0 - speech_irm_prediction
107
 
 
 
 
108
  # speech_spec = mix_spec_complex * mask_speech
109
  noise_spec = mix_spec_complex * mask_noise
110
 
 
105
  mask_speech = speech_irm_prediction
106
  mask_noise = 1.0 - speech_irm_prediction
107
 
108
+ print(f"mix_spec_complex: {mix_spec_complex.shape}")
109
+ print(f"mask_noise: {mask_noise.shape}")
110
+
111
  # speech_spec = mix_spec_complex * mask_speech
112
  noise_spec = mix_spec_complex * mask_noise
113