Skip to content

Commit 725c74e

Browse files
committed
Correct rnnt calling arguments
1 parent 77d4421 commit 725c74e

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/torchaudio/functional/functional.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1819,13 +1819,13 @@ def _rnnt_loss(
18191819
blank = logits.shape[-1] + blank
18201820

18211821
costs = RnntLoss.apply(
1822-
logits=logits,
1823-
targets=targets,
1824-
logit_lengths=logit_lengths,
1825-
target_lengths=target_lengths,
1826-
blank=blank,
1827-
clamp=clamp,
1828-
fused_log_softmax=fused_log_softmax,
1822+
logits,
1823+
targets,
1824+
logit_lengths,
1825+
target_lengths,
1826+
blank,
1827+
clamp,
1828+
fused_log_softmax
18291829
)
18301830

18311831
if reduction == "mean":

0 commit comments

Comments
 (0)