From 02bc776c35b0723f5eac842669cd2e34ab08d702 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 7 Apr 2021 16:04:48 +0200
Subject: [PATCH] prevenet grad in TorchSTFT

---
 TTS/vocoder/layers/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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