File size: 554 Bytes
22f52a4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
trainer:
logger: false
callbacks:
- class_path: SearchCallback
init_args:
index_dir: ./index
search_config:
class_path: SparseSearchConfig
init_args:
k: 10
model:
class_path: lightning_ir.BiEncoderModule
init_args:
model_name_or_path: webis/bert-bi-encoder
evaluation_metrics:
- nDCG@10
data:
class_path: lightning_ir.LightningIRDataModule
init_args:
inference_datasets:
- class_path: QueryDataset
init_args:
doc_dataset: msmarco-passage/trec-dl-2019/judged |