From 79b877f037dba7ff72add51131b96d3508dca68b Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Fri, 16 Jun 2023 14:52:39 +0800
Subject: [PATCH] CTC loss return tensor

---
 egs/librispeech/ASR/zipformer/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/zipformer/model.py b/egs/librispeech/ASR/zipformer/model.py
index 0c3ea6a86..a1367c068 100644
--- a/egs/librispeech/ASR/zipformer/model.py
+++ b/egs/librispeech/ASR/zipformer/model.py
@@ -353,6 +353,6 @@ class AsrModel(nn.Module):
                 target_lengths=y_lens,
             )
         else:
-            ctc_loss = 0
+            ctc_loss = torch.zeros(0)
 
         return simple_loss, pruned_loss, ctc_loss