提交 eb3f9aa6 编写于 作者: X xj.lin

fix


Former-commit-id: 7bb0559f007ef32b8331be8d4947ebe7595714b3
上级 8b1b54f0
...@@ -191,7 +191,9 @@ server::KnowhereError write_index(VecIndexPtr index, const std::string &location ...@@ -191,7 +191,9 @@ server::KnowhereError write_index(VecIndexPtr index, const std::string &location
// TODO(linxj): redo here. // TODO(linxj): redo here.
void AutoGenParams(const IndexType &type, const long &size, zilliz::knowhere::Config &cfg) { void AutoGenParams(const IndexType &type, const long &size, zilliz::knowhere::Config &cfg) {
if (!cfg.contains("nlist")) { cfg["nlist"] = int(size / 1000000.0 * 16384); } auto nlist = cfg.get_with_default("nlist", 0);
if (int(size/1000000.0) * nlist == 0) { cfg["nlist"] = int(size / 1000000.0 * 16384); }
if (!cfg.contains("gpu_id")) { cfg["gpu_id"] = int(0); } if (!cfg.contains("gpu_id")) { cfg["gpu_id"] = int(0); }
if (!cfg.contains("metric_type")) { cfg["metric_type"] = "L2"; } if (!cfg.contains("metric_type")) { cfg["metric_type"] = "L2"; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册