update re-rank example config

webis-de · Jan 27, 2025 · df6b5d9 · df6b5d9
1 parent cec674c
commit df6b5d9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/examples/re-rank.yaml b/configs/examples/re-rank.yaml
@@ -4,7 +4,7 @@ trainer:
     init_args:
       save_dir: ./re-ranked-runs
 model:
-  class_path: lightning_ir.BiEncoderModule
+  class_path: lightning_ir.CrossEncoderModule
   init_args:
     model_name_or_path: webis/monoelectra-base
     evaluation_metrics:
@@ -15,8 +15,8 @@ data:
     inference_datasets:
     - class_path: lightning_ir.RunDataset
       init_args:
-        doc_dataset: ./runs/msmarco-passage-trec-dl-2019-judged.run
+        run_path_or_id: ./runs/msmarco-passage-trec-dl-2019-judged.run
     - class_path: lightning_ir.RunDataset
       init_args:
-        doc_dataset: ./runs/msmarco-passage-trec-dl-2020-judged.run
+        run_path_or_id: ./runs/msmarco-passage-trec-dl-2020-judged.run
     inference_batch_size: 4