提交 48ed10ab 编写于 作者: Z zirui.chen

fix problem of hardcore nprobe

上级 be0bf607
......@@ -18,6 +18,7 @@
#include "wrapper/ConfAdapter.h"
#include "knowhere/index/vector_index/helpers/IndexParameter.h"
#include "utils/Log.h"
#include "WrapperException.h"
#include <cmath>
#include <memory>
......@@ -146,8 +147,10 @@ IVFPQConfAdapter::MatchSearch(const TempMetaConf& metaconf, const IndexType& typ
auto conf = std::make_shared<knowhere::IVFPQCfg>();
conf->k = metaconf.k;
if (metaconf.nprobe <= 0)
conf->nprobe = 16; // hardcode here
if (metaconf.nprobe <= 0){
WRAPPER_LOG_ERROR << "The nprobe of PQ is wrong!";
throw WrapperException("The nprobe of PQ is wrong!");
}
else
conf->nprobe = metaconf.nprobe;
......
......@@ -51,6 +51,7 @@ class KnowhereWrapperTest
tempconf.size = nb;
tempconf.dim = dim;
tempconf.k = k;
tempconf.nprobe = 16;
index_ = GetVecIndexFactory(index_type);
conf = ParamGenerator::GetInstance().GenBuild(index_type, tempconf);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册