雾聪
2024-03-29 9ba0dbd98bf69c830dfcfde8f109a400cb65e4e5
runtime/onnxruntime/src/paraformer-torch.cpp
@@ -281,13 +281,18 @@
        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;