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