From e9f6703350fc6616b06c0e60944f6359a329e214 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期三, 15 三月 2023 15:29:31 +0800
Subject: [PATCH] update
---
funasr/bin/eend_ola_inference.py | 2 --
funasr/models/frontend/wav_frontend.py | 5 +++++
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/eend_ola_inference.py b/funasr/bin/eend_ola_inference.py
index fbcfc7d..2887b37 100755
--- a/funasr/bin/eend_ola_inference.py
+++ b/funasr/bin/eend_ola_inference.py
@@ -227,8 +227,6 @@
output_writer = open("{}/result.txt".format(output_path), "w")
result_list = []
for keys, batch in loader:
- logger.info("keys: {}".format(keys))
- logger.info("batch: {}".format(batch))
assert isinstance(batch, dict), type(batch)
assert all(isinstance(s, str) for s in keys), keys
_bs = len(next(iter(batch.values())))
diff --git a/funasr/models/frontend/wav_frontend.py b/funasr/models/frontend/wav_frontend.py
index f61d7dd..8e17102 100644
--- a/funasr/models/frontend/wav_frontend.py
+++ b/funasr/models/frontend/wav_frontend.py
@@ -11,6 +11,8 @@
import funasr.models.frontend.eend_ola_feature as eend_ola_feature
from funasr.models.frontend.abs_frontend import AbsFrontend
+from modelscope.utils.logger import get_logger
+logger = get_logger()
def load_cmvn(cmvn_file):
with open(cmvn_file, 'r', encoding='utf-8') as f:
@@ -485,6 +487,9 @@
batch_size = input.size(0)
feats = []
feats_lens = []
+ logger.info("batch_size: {}".format(batch_size))
+ logger.info("input: {}".format(input))
+ logger.info("input_lengths: {}".format(input_lengths))
for i in range(batch_size):
waveform_length = input_lengths[i]
waveform = input[i][:waveform_length]
--
Gitblit v1.9.1