From 93a02feda1848531a48a1794a8de12a962c84a8e Mon Sep 17 00:00:00 2001
From: Xingchen Song(宋星辰) <xingchensong1996@163.com>
Date: 星期五, 07 六月 2024 23:10:13 +0800
Subject: [PATCH] [fix] fix empty asr result (#1794)
---
funasr/bin/compute_audio_cmvn.py | 11 +++++++----
1 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/funasr/bin/compute_audio_cmvn.py b/funasr/bin/compute_audio_cmvn.py
index 91bb2ac..e8ced51 100644
--- a/funasr/bin/compute_audio_cmvn.py
+++ b/funasr/bin/compute_audio_cmvn.py
@@ -52,7 +52,7 @@
frontend=frontend,
tokenizer=None,
is_training=False,
- **kwargs.get("dataset_conf")
+ **kwargs.get("dataset_conf"),
)
# dataloader
@@ -68,11 +68,14 @@
dataset_train, collate_fn=dataset_train.collator, **batch_sampler_train
)
- iter_stop = int(kwargs.get("scale", 1.0) * len(dataloader_train))
-
total_frames = 0
for batch_idx, batch in enumerate(dataloader_train):
- if batch_idx >= iter_stop:
+ iter_stop = int(kwargs.get("scale", -1.0) * len(dataloader_train))
+ log_step = iter_stop // 100
+ if batch_idx % log_step == 0:
+ logging.info(f"prcessed: {batch_idx}/{iter_stop}")
+ if batch_idx >= iter_stop and iter_stop > 0.0:
+ logging.info(f"prcessed: {iter_stop}/{iter_stop}")
break
fbank = batch["speech"].numpy()[0, :, :]
--
Gitblit v1.9.1