Remove gradient variable from RNNTL Python code (pytorch#1616)

yangarbiter · Jul 8, 2021 · d6ae55c · d6ae55c
1 parent 7f591c6
commit d6ae55c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,