Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Feb 25

Commit

877f661

1 Parent(s): efe955e

update

Files changed (1) hide show

toolbox/torchaudio/models/clean_unet/loss.py CHANGED Viewed

@@ -16,7 +16,7 @@ def stft(x, fft_size, hop_size, win_length, window):
     :return: Magnitude spectrogram (B, #frames, fft_size // 2 + 1).
     """
-    x_stft = torch.stft(x, fft_size, hop_size, win_length, window)
     return x_stft.abs()

     :return: Magnitude spectrogram (B, #frames, fft_size // 2 + 1).
     """
+    x_stft = torch.stft(x, fft_size, hop_size, win_length, window, return_complex=True)
     return x_stft.abs()