Browse Source

stft.py: fix filter winlength error

master
rafaelvalle 6 years ago
parent
commit
e314bb4cd0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      stft.py

+ 1
- 1
stft.py View File

@ -61,7 +61,7 @@ class STFT(torch.nn.Module):
np.linalg.pinv(scale * fourier_basis).T[:, None, :]) np.linalg.pinv(scale * fourier_basis).T[:, None, :])
if window is not None: if window is not None:
assert(win_length >= filter_length)
assert(filter_length >= win_length)
# get window and zero center pad it to filter_length # get window and zero center pad it to filter_length
fft_window = get_window(window, win_length, fftbins=True) fft_window = get_window(window, win_length, fftbins=True)
fft_window = pad_center(fft_window, filter_length) fft_window = pad_center(fft_window, filter_length)

Loading…
Cancel
Save