From 28719e181ad700839c6a344c5ecee12b3559b016 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期日, 11 十二月 2022 22:03:09 +0800
Subject: [PATCH] paraformer batch padding
---
funasr/bin/asr_inference_paraformer.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/asr_inference_paraformer.py b/funasr/bin/asr_inference_paraformer.py
index 15a37f7..9dcd0b8 100755
--- a/funasr/bin/asr_inference_paraformer.py
+++ b/funasr/bin/asr_inference_paraformer.py
@@ -441,7 +441,7 @@
"decoding, feature length: {}, forward_time: {:.4f}, rtf: {:.4f}".
format(length, forward_time, 100 * forward_time / (length*lfr_factor)))
- for batch_id in range(len(results)):
+ for batch_id in range(_bs):
result = [results[batch_id][:-2]]
key = keys[batch_id]
--
Gitblit v1.9.1