Remove in-place operations from GriffinLim for backprop (#730)

bc1df488 · Peter Massey-Plantinga · GitHub · b17da7a4 · bc1df488
Unverified Commit bc1df488 authored Jun 18, 2020 by Peter Massey-Plantinga Committed by GitHub Jun 18, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchaudio/functional.py torchaudio/functional.py +1 -1

No files found.
--- a/torchaudio/functional.py
+++ b/torchaudio/functional.py
@@ -265,7 +265,7 @@ def griffinlim(
        angles = rebuilt
        if momentum:
            angles = angles - tprev.mul_(momentum / (1 + momentum))
-        angles = angles.div_(complex_norm(angles).add_(1e-16).unsqueeze(-1).expand_as(angles))
+        angles = angles.div(complex_norm(angles).add(1e-16).unsqueeze(-1).expand_as(angles))

    # Return the final phase estimates
    waveform = istft(specgram * angles,