We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 77d4421 commit 725c74eCopy full SHA for 725c74e
src/torchaudio/functional/functional.py
@@ -1819,13 +1819,13 @@ def _rnnt_loss(
1819
blank = logits.shape[-1] + blank
1820
1821
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,
+ logits,
+ targets,
+ logit_lengths,
+ target_lengths,
+ blank,
+ clamp,
+ fused_log_softmax
1829
)
1830
1831
if reduction == "mean":
0 commit comments