milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix problem of hardcore nprobe

pull/324/head
zirui.chen 2019-11-15 18:54:19 +08:00
parent be0bf607ca
commit 48ed10abdc
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core/src/wrapper/ConfAdapter.cpp
View File

@ -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;

1
core/unittest/wrapper/test_wrapper.cpp
View File

@ -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);