From 3e333c0abf31825e84d9673faf5e77601ced1112 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期四, 16 三月 2023 16:49:03 +0800
Subject: [PATCH] space between tokens
---
funasr/models_transducer/espnet_transducer_model_unified.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models_transducer/espnet_transducer_model_unified.py b/funasr/models_transducer/espnet_transducer_model_unified.py
index efe3f4e..6df86f8 100644
--- a/funasr/models_transducer/espnet_transducer_model_unified.py
+++ b/funasr/models_transducer/espnet_transducer_model_unified.py
@@ -455,7 +455,8 @@
gather=True,
)
- if not self.training and (self.report_cer or self.report_wer):
+ #if not self.training and (self.report_cer or self.report_wer):
+ if self.report_cer or self.report_wer:
if self.error_calculator is None:
self.error_calculator = ErrorCalculator(
self.decoder,
@@ -468,7 +469,6 @@
)
cer_transducer, wer_transducer = self.error_calculator(encoder_out, target)
-
return loss_transducer, cer_transducer, wer_transducer
return loss_transducer, None, None
--
Gitblit v1.9.1