update TorchSTFT to enable melspec

2021-04-06 11:01:39 +02:00 · 2021-04-06 11:01:39 +02:00 · ff07c5f5e3
parent 4a5b1d4ac2
commit ff07c5f5e3
1 changed files with 1 additions and 0 deletions
--- a/TTS/vocoder/layers/losses.py
+++ b/TTS/vocoder/layers/losses.py
@ -17,6 +17,7 @@ class TorchSTFT(nn.Module):  # pylint: disable=abstract-method
                 mel_fmax=None,
                 n_mels=80,
                 use_mel=False):
+        """ Torch based STFT operation """
        super(TorchSTFT, self).__init__()
        self.n_fft = n_fft
        self.hop_length = hop_length