funasr/models/e2e_asr_paraformer.py
@@ -236,10 +236,11 @@ loss = self.ctc_weight * loss_ctc + (1 - self.ctc_weight) * loss_att + loss_pre * self.predictor_weight if self.use_1st_decoder_loss and pre_loss_att is not None: loss = loss + pre_loss_att loss = loss + (1 - self.ctc_weight) * pre_loss_att # Collect Attn branch stats stats["loss_att"] = loss_att.detach() if loss_att is not None else None stats["pre_loss_att"] = pre_loss_att.detach() if pre_loss_att is not None else None stats["acc"] = acc_att stats["cer"] = cer_att stats["wer"] = wer_att
