From d667dc365b3259179c9d54bd32a1bb2bd8afa4f0 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Fri, 16 Jun 2023 15:04:41 +0800
Subject: [PATCH] Fix for diagnostic (#1135)

* CTC loss return tensor

* Update model.py
---
 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 0c3ea6a86..b541ee697 100644
--- a/egs/librispeech/ASR/zipformer/model.py
+++ b/egs/librispeech/ASR/zipformer/model.py
@@ -340,8 +340,8 @@ class AsrModel(nn.Module):
                 lm_scale=lm_scale,
             )
         else:
-            simple_loss = 0
-            pruned_loss = 0
+            simple_loss = torch.empty(0)
+            pruned_loss = torch.empty(0)
 
         if self.use_ctc:
             # Compute CTC loss
@@ -353,6 +353,6 @@ class AsrModel(nn.Module):
                 target_lengths=y_lens,
             )
         else:
-            ctc_loss = 0
+            ctc_loss = torch.empty(0)
 
         return simple_loss, pruned_loss, ctc_loss