set requires_grad=False

2026-09-09 11:16:00 +08:00 · 2021-01-16 19:46:04 +01:00
parent c96f7a2614
commit b2b4828f17
1 changed files with 2 additions and 1 deletions
@@ -11,7 +11,8 @@ class TorchSTFT(nn.Module):
        self.n_fft = n_fft
        self.hop_length = hop_length
        self.win_length = win_length
-        self.window = nn.Parameter(getattr(torch, window)(win_length))
+        self.window = nn.Parameter(getattr(torch, window)(win_length),
+                                   requires_grad=False)

    def __call__(self, x):
        # B x D x T x 2