From 97ed4fada4770a11ee803e72020d793fcc5251c0 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 26 四月 2023 13:28:43 +0800
Subject: [PATCH] Merge pull request #423 from alibaba-damo-academy/dev_aky
---
funasr/models/e2e_asr_transducer.py | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/funasr/models/e2e_asr_transducer.py b/funasr/models/e2e_asr_transducer.py
index 657dd75..f8ba0f0 100644
--- a/funasr/models/e2e_asr_transducer.py
+++ b/funasr/models/e2e_asr_transducer.py
@@ -386,7 +386,7 @@
if not self.training and (self.report_cer or self.report_wer):
if self.error_calculator is None:
- from espnet2.asr_transducer.error_calculator import ErrorCalculator
+ from funasr.modules.e2e_asr_common import ErrorCalculatorTransducer as ErrorCalculator
self.error_calculator = ErrorCalculator(
self.decoder,
@@ -398,7 +398,7 @@
report_wer=self.report_wer,
)
- cer_transducer, wer_transducer = self.error_calculator(encoder_out, target)
+ cer_transducer, wer_transducer = self.error_calculator(encoder_out, target, t_len)
return loss_transducer, cer_transducer, wer_transducer
@@ -889,6 +889,8 @@
if not self.training and (self.report_cer or self.report_wer):
if self.error_calculator is None:
+ from funasr.modules.e2e_asr_common import ErrorCalculatorTransducer as ErrorCalculator
+
self.error_calculator = ErrorCalculator(
self.decoder,
self.joint_network,
@@ -899,7 +901,7 @@
report_wer=self.report_wer,
)
- cer_transducer, wer_transducer = self.error_calculator(encoder_out, target)
+ cer_transducer, wer_transducer = self.error_calculator(encoder_out, target, t_len)
return loss_transducer, cer_transducer, wer_transducer
return loss_transducer, None, None
--
Gitblit v1.9.1