prevenet grad in TorchSTFT

2021-04-07 16:04:48 +02:00 · 2021-04-07 16:04:48 +02:00 · 02bc776c35
parent cf44624df8
commit 02bc776c35
1 changed files with 1 additions and 1 deletions
--- a/TTS/vocoder/layers/losses.py
+++ b/TTS/vocoder/layers/losses.py
@ -33,7 +33,7 @@ class TorchSTFT(nn.Module):  # pylint: disable=abstract-method
        self.mel_basis = None
        if use_mel:
            self._build_mel_basis()
-
+    @torch.no_grad()
    def __call__(self, x):
        """Compute spectrogram frames by torch based stft.