stft.py: fix filter winlength error

clean_slate
rafaelvalle 2018-11-25 22:33:52 -08:00
parent 4af4ccb135
commit e314bb4cd0
1 changed files with 1 additions and 1 deletions

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)