Hugging Face's logo

webis
/
bert-bi-encoder

Safetensors
bi-encoder
Model card Files Community
bert-bi-encoder
File size: 574 Bytes 
be1a6a8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 

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