| trainer: | |
| logger: false | |
| callbacks: | |
| - class_path: lightning_ir.IndexCallback | |
| init_args: | |
| index_dir: ./index | |
| index_config: | |
| class_path: FaissIVFPQIndexConfig | |
| init_args: | |
| num_centroids: 262144 | |
| num_subquantizers: 16 | |
| n_bits: 8 | |
| model: | |
| class_path: lightning_ir.BiEncoderModule | |
| init_args: | |
| model_name_or_path: webis/bert-bi-encoder | |
| data: | |
| class_path: lightning_ir.LightningIRDataModule | |
| init_args: | |
| num_workers: 1 | |
| inference_batch_size: 256 | |
| inference_datasets: | |
| - class_path: DocDataset | |
| init_args: | |
| doc_dataset: msmarco-passage |