| 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 |
