update repo

This commit is contained in:
嘉渊 2023-05-17 19:55:59 +08:00
parent 701022837a
commit bd4bf9928d

View File

@ -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