Skip to content

Commit c78b18c

Browse files
mdouzefacebook-github-bot
authored andcommitted
remove setNumProbes (facebookresearch#2797)
Summary: Pull Request resolved: facebookresearch#2797 This is the last code instance of setNumProbes Removing because some people still seem run into errors due to this. Reviewed By: algoriddle Differential Revision: D44421600 fbshipit-source-id: fbc1a9d49a0175ddf24c32dab5c1bdb5f1bbbac6
1 parent 4012a78 commit c78b18c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchs/bench_gpu_sift1m.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,8 @@
8585

8686
for lnprobe in range(10):
8787
nprobe = 1 << lnprobe
88-
index.setNumProbes(nprobe)
88+
index.nprobe
89+
index.nprobe = nprobe
8990
t, r = evaluate(index, xq, gt, 100)
9091

9192
print("nprobe=%4d %.3f ms recalls= %.4f %.4f %.4f" % (nprobe, t, r[1], r[10], r[100]))

0 commit comments

Comments
 (0)