Fix for Pytorch warning message

Warning:
(...)/torch/functional.py:650: UserWarning: stft with return_complex=False is deprecated. In a future pytorch release, stft will return complex tensors for all inputs, and return_complex=False will raise an error.
This commit is contained in:
r3dacted42 2023-11-27 19:19:02 +05:30
parent 11ec9f7471
commit 4a1a8f831a
1 changed files with 1 additions and 1 deletions

View File

@ -129,7 +129,7 @@ class TorchSTFT(nn.Module): # pylint: disable=abstract-method
pad_mode="reflect", # compatible with audio.py
normalized=self.normalized,
onesided=True,
return_complex=False,
return_complex=True,
)
M = o[:, :, :, 0]
P = o[:, :, :, 1]