Fix document for forced_align method (#3748)

The lengths of targets and log_probs should be reversed.
pytorch · Feb 16, 2024 · b7b7b5d · b7b7b5d
1 parent 5286f9f
commit b7b7b5d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/libtorchaudio/forced_align/cpu/compute.cpp b/src/libtorchaudio/forced_align/cpu/compute.cpp
@@ -42,9 +42,9 @@ void forced_align_impl(
   }
   TORCH_CHECK(
       T >= L + R,
-      "targets length is too long for CTC. Found targets length: ",
+      "targets length is too long for CTC. Found log_probs length: ",
       T,
-      ", log_probs length: ",
+      ", targets length: ",
       L,
       ", and number of repeats: ",
       R);

diff --git a/src/libtorchaudio/forced_align/gpu/compute.cu b/src/libtorchaudio/forced_align/gpu/compute.cu
@@ -160,9 +160,9 @@ void forced_align_impl(
   }
   TORCH_CHECK(
       T >= L + R,
-      "targets length is too long for CTC. Found targets length: ",
+      "targets length is too long for CTC. Found log_probs length: ",
       T,
-      ", log_probs length: ",
+      ", targets length: ",
       L,
       ", and number of repeats: ",
       R);