From d295d5de97f7eb20acdf01c4a093a8dc6a9b3384 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sat, 10 Apr 2021 19:43:57 +0200 Subject: [PATCH] remove torch.no_grad from TorchSTFT --- TTS/vocoder/layers/losses.py | 1 - 1 file changed, 1 deletion(-) 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.