yrshi commited on
Commit
81fc00b
·
1 Parent(s): 3b64038

rename somefiles

Browse files
Files changed (1) hide show
  1. retrieval_launch.sh +12 -0
retrieval_launch.sh ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ file_path=./data
3
+ index_file=$file_path/e5_Flat.index
4
+ corpus_file=$file_path/wiki-18.jsonl
5
+ retriever=intfloat/e5-base-v2
6
+
7
+ export CUDA_VISIBLE_DEVICES="1,3"
8
+ python search_r1/search/retrieval_server.py --index_path $index_file \
9
+ --corpus_path $corpus_file \
10
+ --topk 3 \
11
+ --retriever_model $retriever \
12
+ --faiss_gpu False