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
|
pad_mode="reflect", # compatible with audio.py
|
||||||
normalized=self.normalized,
|
normalized=self.normalized,
|
||||||
onesided=True,
|
onesided=True,
|
||||||
return_complex=False,
|
return_complex=True,
|
||||||
)
|
)
|
||||||
M = o[:, :, :, 0]
|
M = o[:, :, :, 0]
|
||||||
P = o[:, :, :, 1]
|
P = o[:, :, :, 1]
|
||||||
|
|
Loading…
Reference in New Issue