fix memory leak

KentoNishi · Jun 21, 2021 · abe436b · abe436b
1 parent d95a218
commit abe436b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_pitch_shift/main.py b/torch_pitch_shift/main.py
@@ -114,6 +114,7 @@ def __call__(
         ).to(input.device)
         output = stretcher(output)
         output = torch.istft(output[0], self._n_fft)
+        del resampler, stretcher
         if output.shape[1] >= input.shape[1]:
             return output[:, : (input.shape[1])]
         return pad(output, pad=(0, input.shape[1] - output.shape[1], 0, 0))