fix warning

This commit is contained in:
James Betker 2022-05-02 16:36:02 -06:00
parent ee24d3ee4b
commit 4836e1f792

View File

@ -116,7 +116,7 @@ class STFT(torch.nn.Module):
assert(filter_length >= win_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, size=filter_length)
fft_window = torch.from_numpy(fft_window).float() fft_window = torch.from_numpy(fft_window).float()
# window the bases # window the bases