speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix func Forward

Browse Source
This commit is contained in:
雾聪 2024-03-29 17:24:59 +08:00
parent b2d549b954
commit 9ba0dbd98b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
runtime/onnxruntime/src/paraformer-torch.cpp
View File

@ -281,13 +281,18 @@ std::vector<std::string> ParaformerTorch::Forward(float** din, int* len, bool in
if(asr_feats.size() != 0){
LfrCmvn(asr_feats);
}
feats_batch.emplace_back(asr_feats);
int32_t num_frames = asr_feats.size() / feature_dim;
paraformer_length.emplace_back(num_frames);
if(max_size < asr_feats.size()){
max_size = asr_feats.size();
max_frames = num_frames;
}
std::vector<float> flattened;
for (const auto& sub_vector : asr_feats) {
flattened.insert(flattened.end(), sub_vector.begin(), sub_vector.end());
}
feats_batch.emplace_back(flattened);
}
torch::NoGradGuard no_grad;