From 04d4152575c11d14d00657f6d24090dece96c68e Mon Sep 17 00:00:00 2001 From: PeterGriffinJin Date: Mon, 7 Apr 2025 18:35:41 +0000 Subject: [PATCH] add indexing for ANN and bm25 --- search_r1/search/build_index.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/search_r1/search/build_index.sh b/search_r1/search/build_index.sh index 5a892ec..05556a3 100644 --- a/search_r1/search/build_index.sh +++ b/search_r1/search/build_index.sh @@ -4,6 +4,8 @@ save_dir=/the/path/to/save/index retriever_name=e5 # this is for indexing naming retriever_model=intfloat/e5-base-v2 +# change faiss_type to HNSW32/64/128 for ANN indexing +# change retriever_name to bm25 for BM25 indexing CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python index_builder.py \ --retrieval_method $retriever_name \ --model_path $retriever_model \