Skip to content

GitLab

Unverified Commit d6ae55c6 authored by Caroline Chen's avatar Caroline Chen Committed by GitHub
Browse files

Remove gradient variable from RNNTL Python code (#1616)

parent 7f591c62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchaudio/prototype/rnnt_loss.py
View file @ d6ae55c6
...@@ -53,7 +53,7 @@ def rnnt_loss( ...@@ -53,7 +53,7 @@ def rnnt_loss(
if blank < 0: # reinterpret blank index if blank < 0. if blank < 0: # reinterpret blank index if blank < 0.
blank = logits.shape[-1] + blank blank = logits.shape[-1] + blank
costs, gradients = torch.ops.torchaudio.rnnt_loss( costs, _ = torch.ops.torchaudio.rnnt_loss(
logits=logits, logits=logits,
targets=targets, targets=targets,
logit_lengths=logit_lengths, logit_lengths=logit_lengths,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment