mirror of https://github.com/coqui-ai/TTS.git
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:
parent
11ec9f7471
commit
4a1a8f831a
|
@ -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]
|
||||
|
|
Loading…
Reference in New Issue