update

examples/spectrum_dfnet_aishell/step_3_evaluation.py

@@ -213,14 +213,16 @@ def main():
         speech_spec: torch.Tensor = stft_power.forward(speech_wave)
         mix_spec: torch.Tensor = stft_power.forward(mix_wave)
 
-        noise_spec = noise_spec[:, :-1, :]
-        speech_spec = speech_spec[:, :-1, :]
-        mix_spec = mix_spec[:, :-1, :]
-
         speech_spec_complex: torch.Tensor = stft_complex.forward(speech_wave)
         mix_spec_complex: torch.Tensor = stft_complex.forward(mix_wave)
         # mix_spec_complex shape: [batch_size, freq_dim (257), time_steps, 2]
 
+        noise_spec = noise_spec[:, :-1, :]
+        speech_spec = speech_spec[:, :-1, :]
+        mix_spec = mix_spec[:, :-1, :]
+        speech_spec_complex = speech_spec_complex[:, :-1, :]
+        mix_spec_complex = mix_spec_complex[:, :-1, :]
+
         speech_irm = speech_spec / (noise_spec + speech_spec)
         speech_irm = torch.pow(speech_irm, 1.0)