嘉渊
2023-05-18 a7002c3285dc6c6db4bac2666b473e0a3a3bdb9b
funasr/models/e2e_asr_paraformer.py
@@ -240,6 +240,7 @@
        # 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