From bd4bf9928df670a7b6a30b750fc6522156f2e739 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Wed, 17 May 2023 19:55:59 +0800
Subject: [PATCH] update repo

---
 funasr/models/e2e_asr_paraformer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py
index 060442cc9..924127117 100644
--- a/funasr/models/e2e_asr_paraformer.py
+++ b/funasr/models/e2e_asr_paraformer.py
@@ -240,6 +240,7 @@ class Paraformer(FunASRModel):
 
         # 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