update TorchSTFT to enable melspec

2026-09-09 11:16:00 +08:00 · 2021-04-08 11:25:29 +02:00
parent 4a5b1d4ac2
commit ff07c5f5e3
1 changed files with 1 additions and 0 deletions
@@ -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