EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Support audio tagging using zipformer (#747)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-04-10 14:47:06 +08:00
committed by GitHub
parent c9ae7595d5
commit f20291cadc
30 changed files with 927 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sherpa-onnx/csrc/math.h
View File

@@ -97,8 +97,8 @@ void LogSoftmax(T *in, int32_t w, int32_t h) {
}
template <typename T>
void SubtractBlank(T *in, int32_t w, int32_t h,
int32_t blank_idx, float blank_penalty) {
void SubtractBlank(T *in, int32_t w, int32_t h, int32_t blank_idx,
float blank_penalty) {
for (int32_t i = 0; i != h; ++i) {
in[blank_idx] -= blank_penalty;
in += w;
@@ -116,8 +116,7 @@ std::vector<int32_t> TopkIndex(const T *vec, int32_t size, int32_t topk) {
});
int32_t k_num = std::min<int32_t>(size, topk);
std::vector<int32_t> index(vec_index.begin(), vec_index.begin() + k_num);
return index;
return {vec_index.begin(), vec_index.begin() + k_num};
}
} // namespace sherpa_onnx