| trainer: | |
| logger: false | |
| callbacks: | |
| - class_path: SearchCallback | |
| init_args: | |
| index_dir: ./index | |
| use_gpu: false | |
| search_config: | |
| class_path: FaissSearchConfig | |
| 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 |