add basic configs

configs/callbacks/rank.yaml → configs/callbacks/re-rank.yaml

@@ -1,5 +1,5 @@
 trainer:
   callbacks+:
-  - class_path: RankCallback
+  - class_path: ReRankCallback
     init_args:
       save_dir: ./runs

configs/data/run-dataset.yaml

@@ -0,0 +1,12 @@
+data:
+  class_path: LightningIRDataModule
+  init_args:
+    num_workers: 1
+    inference_batch_size: 4
+    inference_datasets:
+      - class_path: RunDataset
+        init_args:
+          run_path_or_id: null
+          depth: 100
+          sample_size: 100
+          sampling_strategy: top

configs/models/bi-encoder.yaml

@@ -0,0 +1,4 @@
+model:
+  class_path: BiEncoderModule
+  init_args:
+    model_name_or_path: null

configs/models/colbert.yaml

@@ -0,0 +1,19 @@
+model:
+  class_path: BiEncoderModule
+  init_args:
+    model_name_or_path: ./tests/data/test_model
+    config:
+      class_path: ColConfig
+      init_args:
+        similarity_function: dot
+        query_aggregation_function:  sum
+        query_expansion: true
+        query_length: 32
+        doc_length: 256
+        normalize: true
+        add_marker_tokens: true
+        embedding_dim: 128
+        linear_bias: false
+    loss_functions:
+      - class_path: RankNet
+

configs/models/cross-encoder.yaml

@@ -0,0 +1,4 @@
+model:
+  class_path: CrossEncoderModule
+  init_args:
+    model_name_or_path: null