From a063afb349212e20ab1d406e984f17bb247e6592 Mon Sep 17 00:00:00 2001 From: Mistmoon <2070620600@qq.com> Date: Mon, 7 Jul 2025 10:15:44 +0800 Subject: [PATCH] Update egs/aishell/ASR/zipformer/train.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --- egs/aishell/ASR/zipformer/train.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/egs/aishell/ASR/zipformer/train.py b/egs/aishell/ASR/zipformer/train.py index e19bd9282..3104665b0 100755 --- a/egs/aishell/ASR/zipformer/train.py +++ b/egs/aishell/ASR/zipformer/train.py @@ -856,7 +856,10 @@ def compute_loss( supervision_segments=supervision_segments, time_warp_factor=params.spec_aug_time_warp_factor, ) - simple_loss, pruned_loss, ctc_loss, _, cr_loss = losses[:5] + if params.use_ctc: + simple_loss, pruned_loss, ctc_loss, _, cr_loss = losses[:5] + else: + simple_loss, pruned_loss = losses[:2] loss = 0.0