From 2dca94e3af053e6b28592868fae0ab7736cf8c2a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 10 十二月 2022 19:12:20 +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