Allow setting nprobe for IndexIDMap->IndexIVF (#673)

There we go

Allow setting nprobe for IndexIDMap->IndexIVF (#673)
There we go
eb6c4cc9 · Bonnie Varghese · Matthijs Douze · f417a536 · eb6c4cc9
Commit eb6c4cc9 authored Jan 09, 2019 by Bonnie Varghese Committed by Matthijs Douze Jan 09, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

AutoTune.cpp AutoTune.cpp +4 -1

No files found.
--- a/AutoTune.cpp
+++ b/AutoTune.cpp
@@ -488,7 +488,10 @@ void ParameterSpace::set_index_parameter (
    }

    if (name == "nprobe") {
-        if ( DC(IndexIVF)) {
+        if (DC (IndexIDMap)) {
+            set_index_parameter (ix->index, name, val);
+            return;
+        } else if (DC (IndexIVF)) {
            ix->nprobe = int(val);
            return;
        }