-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathance_zero.sh
More file actions
5 lines (5 loc) · 1.28 KB
/
ance_zero.sh
File metadata and controls
5 lines (5 loc) · 1.28 KB
1
2
3
4
5
export PYTHONPATH=${PYTHONPATH}:`pwd`
# CUDA_VISIBLE_DEVICES=0 python drivers/run_convdr_inference.py --model_path=checkpoints/ad-hoc-ance-msmarco --eval_file=datasets/cast-19/eval_topics.jsonl --query=no_res --per_gpu_eval_batch_size=8 --cache_dir=../ann_cache_dir --ann_data_dir=datasets/cast-shared/embeddings --qrels=datasets/cast-19/qrels.tsv --processed_data_dir=datasets/cast-shared --raw_data_dir=datasets/cast-19 --output_dir=results/cast-19 --model_type=rdot_nll --output_query_type=raw --run_tag=ance_zero_19 --use_gpu --max_concat_length=256
# python trec_eval.py --qrel datasets/cast-19/qrels.tsv --run results/cast-19/ance_zero_19.trec --tag cast19
CUDA_VISIBLE_DEVICES=0 python drivers/run_convdr_inference.py --model_path=checkpoints/ad-hoc-ance-msmarco --eval_file=datasets/cast-20/eval_topics.jsonl --query=auto_can --per_gpu_eval_batch_size=8 --cache_dir=../ann_cache_dir --ann_data_dir=datasets/cast-shared/embeddings --qrels=datasets/cast-20/qrels.tsv --processed_data_dir=datasets/cast-shared --raw_data_dir=datasets/cast-20 --output_dir=results/cast-20 --model_type=rdot_nll --output_query_type=raw --run_tag=ance_zero_20 --use_gpu --max_concat_length=512
python trec_eval.py --qrel datasets/cast-20/qrels.tsv --run results/cast-20/ance_zero_20.trec --tag cast20