fix wike datasets config

rapidsai · Oct 31, 2023 · d2ccf2a · d2ccf2a
1 parent 879c744
commit d2ccf2a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml b/python/raft-ann-bench/src/raft-ann-bench/run/conf/datasets.yaml
@@ -85,15 +85,15 @@
   distance: euclidean
 
 - name: wiki_all_1M,
-  base_file: wiki_all_1M/base.88M.fbin,
+  base_file: wiki_all_1M/base.1MM.fbin,
   query_file: wiki_all_1M/queries.fbin,
-  groundtruth_neighbors_file: wiki_all_1M/groundtruth.88M.neighbors.ibin,
+  groundtruth_neighbors_file: wiki_all_1M/groundtruth.1M.neighbors.ibin,
   distance: euclidean
 
 - name: wiki_all_10M,
-  base_file: wiki_all_10M/base.88M.fbin,
+  base_file: wiki_all_10M/base.10M.fbin,
   query_file: wiki_all_10M/queries.fbin,
-  groundtruth_neighbors_file: wiki_all_10M/groundtruth.88M.neighbors.ibin,
+  groundtruth_neighbors_file: wiki_all_10M/groundtruth.10M.neighbors.ibin,
   distance: euclidean
 
 - name: wiki_all_88M,