Fix is_trained in IndexNSGSQ (facebookresearch#3145)

luyuncheng · abhinavdangeti · commit 545cc14b2eb1 · 2024-07-11T19:50:07.000-06:00
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
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;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -309,7 +309,7 @@ IndexNSGSQ::IndexNSGSQ(`
`309`	`309`	`int M,`
`310`	`310`	`MetricType metric)`
`311`	`311`	`: IndexNSG(new IndexScalarQuantizer(d, qtype, metric), M) {`
`312`		`- is_trained = false;`
	`312`	`+ is_trained = this->storage->is_trained;`
`313`	`313`	`own_fields = true;`
`314`	`314`	`}`
`315`	`315`