remove torch.no_grad from TorchSTFT

2021-04-10 19:43:57 +02:00 · 2021-04-10 19:43:57 +02:00 · d295d5de97
parent 5b70da2e3f
commit d295d5de97
1 changed files with 0 additions and 1 deletions
--- a/TTS/vocoder/layers/losses.py
+++ b/TTS/vocoder/layers/losses.py
@ -34,7 +34,6 @@ class TorchSTFT(nn.Module):  # pylint: disable=abstract-method
        if use_mel:
            self._build_mel_basis()

-    @torch.no_grad()
    def __call__(self, x):
        """Compute spectrogram frames by torch based stft.