CAGRA配置

参数值

类型

默认值

说明

proxima.cagra.index.graph_degree

int

64

查询阶段近邻图中节点的邻边数量，越大召回率越好，但性能越差。

proxima.cagra.index.intermediate_graph_degree

int

128

建图阶段构建的近邻图中节点的邻边数量，查询阶段会被裁剪至 proxima.cagra.index.graph_degree，值不小于 proxima.cagra.index.graph_degree，一般不大于1.5倍 proxima.cagra.index.graph_degree。

proxima.cagra.index.mode.fp16

int

0

是否打开fp16量化，建议设置为1（打开），基本不影响召回率。

参数值

类型

默认值

说明

proxima.cagra.search.use_batchable_searcher

int

0

是否打卡攒批，建议设置为1（打开），能够明显提升性能。

proxima.cagra.search.use_gpu_built_index

int

0

是否使用来自CPU构建的图，目前仅支持设置为0（是）。

proxima.cagra.search.itopk_size

int

64

与算法迭代时的候选集大小相关，需要是32的倍数，必须大于等于需要的 topK（如果需要返回 top100，那么这个值必须大于等于 100），越大召回率越好，但性能越差。