in dpr_scale/run_retrieval_multiset.py [0:0]
def get_parser():
parser = argparse.ArgumentParser()
parser.add_argument(
"--ctx_embeddings_dir",
type=str,
default="",
)
parser.add_argument(
"--query_emb_paths",
type=str,
nargs='+',
default=['query_reps.pkl'],
help="list of query emb files"
)
parser.add_argument(
"--questions_jsonl_paths",
type=str,
nargs='+',
default="list of dataset files",
)
parser.add_argument(
"--passages_tsv_path",
type=str,
default="",
)
parser.add_argument(
"--output_json_paths",
type=str,
nargs='+',
default="list of output filenames",
)
return parser