昇腾社区首页
中文
注册

trainCodeBook接口

API定义

void trainCodeBook(const AscendIndexCodeBookInitParams &codeBookInitParams) const;

功能描述

IVFSP码本训练接口。如果训练速度较慢,可能是安装OpenBLAS时限制了使用单线程,可以设置环境变量export OMP_NUM_THREADS=4 进行加速

输入

const AscendIndexCodeBookInitParams &codeBookInitParams:训练码本所需的初始化参数。

输出

返回值

约束说明

参考AscendIndexCodeBookInitParams接口