fix warning
Browse files- tortoise/utils/stft.py +1 -1
tortoise/utils/stft.py
CHANGED
@@ -116,7 +116,7 @@ class STFT(torch.nn.Module):
|
|
116 |
assert(filter_length >= win_length)
|
117 |
# get window and zero center pad it to filter_length
|
118 |
fft_window = get_window(window, win_length, fftbins=True)
|
119 |
-
fft_window = pad_center(fft_window, filter_length)
|
120 |
fft_window = torch.from_numpy(fft_window).float()
|
121 |
|
122 |
# window the bases
|
|
|
116 |
assert(filter_length >= win_length)
|
117 |
# get window and zero center pad it to filter_length
|
118 |
fft_window = get_window(window, win_length, fftbins=True)
|
119 |
+
fft_window = pad_center(fft_window, size=filter_length)
|
120 |
fft_window = torch.from_numpy(fft_window).float()
|
121 |
|
122 |
# window the bases
|