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