AutoRefine / retreival_launch.sh
yrshi's picture
first commit
8b4913f
raw
history blame
428 Bytes
file_path=./data
index_file=$file_path/e5_Flat.index
corpus_file=$file_path/wiki-18.jsonl
retriever=intfloat/e5-base-v2
export CUDA_VISIBLE_DEVICES="1,3"
python search_r1/search/retrieval_server.py --index_path $index_file \
--corpus_path $corpus_file \
--topk 3 \
--retriever_model $retriever