From ff07c5f5e3030219c9e5ac78666259e8932ac516 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 6 Apr 2021 11:01:39 +0200
Subject: [PATCH] update TorchSTFT to enable melspec

---
 TTS/vocoder/layers/losses.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TTS/vocoder/layers/losses.py b/TTS/vocoder/layers/losses.py
index fab70594..ab6f3d35 100644
--- 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