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

fix topk bug which cause ctc score diff

上级 7e334ce8
......@@ -79,11 +79,13 @@ 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--;
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册