Fix small typo in transducer (#1162)

df48ba36 · Vincent QB · GitHub · 463a8b2c · df48ba36
Unverified Commit df48ba36 authored Jan 07, 2021 by Vincent QB Committed by GitHub Jan 07, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchaudio/prototype/transducer.py torchaudio/prototype/transducer.py +2 -2

No files found.
--- a/torchaudio/prototype/transducer.py
+++ b/torchaudio/prototype/transducer.py
@@ -148,8 +148,8 @@ def check_inputs(log_probs, labels, lengths, label_lengths):

    check_dim(log_probs, 4, "log_probs")
    check_dim(labels, 2, "labels")
-    check_dim(lengths, 1, "lenghts")
-    check_dim(label_lengths, 1, "label_lenghts")
+    check_dim(lengths, 1, "lengths")
+    check_dim(label_lengths, 1, "label_lengths")
    max_T = torch.max(lengths)
    max_U = torch.max(label_lengths)
    T, U = log_probs.shape[1:3]