From d93002b5b8e0d12c726566958140e632eb6c9b3a Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 01 四月 2024 11:05:39 +0800
Subject: [PATCH] fix padding
---
runtime/onnxruntime/src/paraformer-torch.cpp | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/runtime/onnxruntime/src/paraformer-torch.cpp b/runtime/onnxruntime/src/paraformer-torch.cpp
index e603e89..9ac0cf7 100644
--- a/runtime/onnxruntime/src/paraformer-torch.cpp
+++ b/runtime/onnxruntime/src/paraformer-torch.cpp
@@ -281,10 +281,10 @@
if(asr_feats.size() != 0){
LfrCmvn(asr_feats);
}
- int32_t num_frames = asr_feats.size() / feature_dim;
+ int32_t num_frames = asr_feats.size();
paraformer_length.emplace_back(num_frames);
- if(max_size < asr_feats.size()){
- max_size = asr_feats.size();
+ if(max_size < asr_feats.size()*feature_dim){
+ max_size = asr_feats.size()*feature_dim;
max_frames = num_frames;
}
--
Gitblit v1.9.1