Fix is_trained in IndexNSGSQ (facebookresearch#3145)

Summary: Same as facebookresearch#3034 When using IndexNSGSQ with fp16, do not require training Pull Request resolved: facebookresearch#3145 Reviewed By: algoriddle Differential Revision: D51615536 Pulled By: mdouze fbshipit-source-id: c6bfbca920be80231d5d0a7290a29f17ea271f6e
blevesearch · Nov 28, 2023 · eb071f8 · eb071f8
1 parent d643c41
commit eb071f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/faiss/IndexNSG.cpp b/faiss/IndexNSG.cpp
@@ -309,7 +309,7 @@ IndexNSGSQ::IndexNSGSQ(
         int M,
         MetricType metric)
         : IndexNSG(new IndexScalarQuantizer(d, qtype, metric), M) {
-    is_trained = false;
+    is_trained = this->storage->is_trained;
     own_fields = true;
 }