AutoRefine / retrieval_launch.sh
yrshi's picture
update some files
de48493
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 retrieval_server.py --index_path $index_file \
--corpus_path $corpus_file \
--topk 3 \
--retriever_model $retriever