Remove gradient variable from RNNTL Python code (#1616)

d6ae55c6 · Caroline Chen · GitHub · 7f591c62 · d6ae55c6
Unverified Commit d6ae55c6 authored Jul 08, 2021 by Caroline Chen Committed by GitHub Jul 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchaudio/prototype/rnnt_loss.py torchaudio/prototype/rnnt_loss.py +1 -1

No files found.
--- a/torchaudio/prototype/rnnt_loss.py
+++ b/torchaudio/prototype/rnnt_loss.py
@@ -53,7 +53,7 @@ def rnnt_loss(
    if blank < 0:  # reinterpret blank index if blank < 0.
        blank = logits.shape[-1] + blank

-    costs, gradients = torch.ops.torchaudio.rnnt_loss(
+    costs, _ = torch.ops.torchaudio.rnnt_loss(
        logits=logits,
        targets=targets,
        logit_lengths=logit_lengths,