Skip to content

Commit 545cc14

Browse files
luyunchengabhinavdangeti
authored andcommitted
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
1 parent 316f9ee commit 545cc14

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

faiss/IndexNSG.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ IndexNSGSQ::IndexNSGSQ(
309309
int M,
310310
MetricType metric)
311311
: IndexNSG(new IndexScalarQuantizer(d, qtype, metric), M) {
312-
is_trained = false;
312+
is_trained = this->storage->is_trained;
313313
own_fields = true;
314314
}
315315

0 commit comments

Comments
 (0)