Skip to content

Commit c1e3840

Browse files
committed
Import rnnt correctly
1 parent 2e2421d commit c1e3840

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/torchaudio/transforms/_transforms.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from torch.nn.parameter import UninitializedParameter
1111

1212
from torchaudio import functional as F
13-
from torchaudio.functional.functional import _rnnt_loss
13+
from torchaudio.functional.functional import rnnt_loss
1414
from torchaudio.functional.functional import (
1515
_apply_sinc_resample_kernel,
1616
_check_convolve_mode,
@@ -1847,7 +1847,7 @@ def forward(
18471847
Tensor: Loss with the reduction option applied. If ``reduction`` is ``"none"``, then size (batch),
18481848
otherwise scalar.
18491849
"""
1850-
return _rnnt_loss(
1850+
return rnnt_loss(
18511851
logits,
18521852
targets,
18531853
logit_lengths,

0 commit comments

Comments
 (0)