-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
42 lines (42 loc) · 1.72 KB
/
config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
{
"METHOD": "PASS",
"MODEL_NAME": "t5",
"T5_MODEL": "model/t5-base/model.ckpt-1004000",
"GPT2_MODEL": "model/gpt-2/",
"DEVICE": "GPU",
"DOC_CONF": {
"CHUNK_SIZE": 128,
"INDEX": "data/doc_rerank/index/lucene-index.msmarco-doc.pos+docvectors+rawdocs",
"QRELS": "data/doc_rerank/qrels/qrels.msmarco-doc.dev.txt",
"QUERY": "data/doc_rerank/query/doc-msmarco-dev-queries.json",
"RANK_FILE": "data/doc_rerank/formatted_run.msmarco-doc.dev.bm25.tuned.txt",
"RESULT": "result/doc_rerank/",
"COLLECTION": [
"data/doc_rerank/collection_jsonl/docs00.json",
"data/doc_rerank/collection_jsonl/docs01.json",
"data/doc_rerank/collection_jsonl/docs02.json",
"data/doc_rerank/collection_jsonl/docs03.json",
"data/doc_rerank/collection_jsonl/docs04.json",
"data/doc_rerank/collection_jsonl/docs05.json",
"data/doc_rerank/collection_jsonl/docs06.json"
]
},
"PASS_CONF": {
"INDEX": "data/pass_rerank/index/lucene-index-msmarco",
"QRELS": "data/pass_rerank/qrels/qrels.dev.small.tsv",
"QUERY": "data/pass_rerank/query/pass-query-dev.small.json",
"RANK_FILE": "data/pass_rerank/pass-dev-qljm-format.small.txt",
"RESULT": "result/pass_rerank/top1000/",
"COLLECTION": [
"data/pass_rerank/collection_jsonl/docs00.json",
"data/pass_rerank/collection_jsonl/docs01.json",
"data/pass_rerank/collection_jsonl/docs02.json",
"data/pass_rerank/collection_jsonl/docs03.json",
"data/pass_rerank/collection_jsonl/docs04.json",
"data/pass_rerank/collection_jsonl/docs05.json",
"data/pass_rerank/collection_jsonl/docs06.json",
"data/pass_rerank/collection_jsonl/docs07.json",
"data/pass_rerank/collection_jsonl/docs08.json"
]
}
}