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