提交 64cb4048 编写于 作者: H Hui Zhang

fix topk bug which cause ctc score diff

上级 7e334ce8
......@@ -79,10 +79,12 @@ void TopK(const std::vector<T>& data,
int cur = values->size() - 1;
while (!pq.empty()) {
const auto& item = pq.top();
pq.pop();
(*values)[cur] = item.first;
(*indices)[cur] = item.second;
// item if reference, must pop here
pq.pop();
cur--;
}
......@@ -93,4 +95,4 @@ template void TopK<float>(const std::vector<float>& data,
std::vector<float>* values,
std::vector<int>* indices);
} // namespace ppspeech
\ No newline at end of file
} // namespace ppspeech
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册