mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
3f0993aebe
commit
006b1d3aaa
|
@ -4,7 +4,7 @@ from torch import nn
|
||||||
from torch.nn import functional as F
|
from torch.nn import functional as F
|
||||||
|
|
||||||
|
|
||||||
class TorchSTFT(nn.Module): # pylint: disable=abstract-method
|
class TorchSTFT():
|
||||||
"""TODO: Merge this with audio.py"""
|
"""TODO: Merge this with audio.py"""
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
n_fft,
|
n_fft,
|
||||||
|
@ -34,7 +34,6 @@ class TorchSTFT(nn.Module): # pylint: disable=abstract-method
|
||||||
if use_mel:
|
if use_mel:
|
||||||
self._build_mel_basis()
|
self._build_mel_basis()
|
||||||
|
|
||||||
@torch.no_grad()
|
|
||||||
def __call__(self, x):
|
def __call__(self, x):
|
||||||
"""Compute spectrogram frames by torch based stft.
|
"""Compute spectrogram frames by torch based stft.
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue