From 6a6cd82b7a9dc7e1a41f9aecb44cea08ad2ad089 Mon Sep 17 00:00:00 2001
From: Teo <teouuenshen@gmail.com>
Date: Thu, 1 Feb 2024 10:01:24 +0900
Subject: [PATCH] fixing torch.ctc err

---
 egs/librispeech/ASR/zipformer/model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/egs/librispeech/ASR/zipformer/model.py b/egs/librispeech/ASR/zipformer/model.py
index f2f86af47..314b39d07 100644
--- a/egs/librispeech/ASR/zipformer/model.py
+++ b/egs/librispeech/ASR/zipformer/model.py
@@ -164,9 +164,9 @@ class AsrModel(nn.Module):
 
         ctc_loss = torch.nn.functional.ctc_loss(
             log_probs=ctc_output.permute(1, 0, 2),  # (T, N, C)
-            targets=targets,
-            input_lengths=encoder_out_lens,
-            target_lengths=target_lengths,
+            targets=targets.to(torch.long),
+            input_lengths=encoder_out_lens.to(torch.long),
+            target_lengths=target_lengths.to(torch.long),
             reduction="sum",
         )
         return ctc_loss