Fix index parameters check of RHNSWPQ (#6950)

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
pull/6842/head
dragondriver 2021-08-03 10:25:25 +08:00 committed by GitHub
parent 47df4a2e57
commit d2767f920c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions

View File

@ -311,7 +311,9 @@ RHNSWPQConfAdapter::CheckTrain(Config& oricfg, const IndexMode mode) {
auto dimension = oricfg[knowhere::meta::DIM].get<int64_t>();
IVFPQConfAdapter::CheckCPUPQParams(dimension, oricfg[knowhere::IndexParams::PQM].get<int64_t>());
if (!IVFPQConfAdapter::CheckCPUPQParams(dimension, oricfg[knowhere::IndexParams::PQM].get<int64_t>())) {
return false;
}
return ConfAdapter::CheckTrain(oricfg, mode);
}