may fix

Parry-Parry · Nov 21, 2024 · e277003 · e277003
1 parent 0a6b1cb
commit e277003
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rankers/modelling/sparse.py b/rankers/modelling/sparse.py
@@ -19,7 +19,7 @@ def __init__(
         model_name_or_path: str = "bert-base-uncased",
         query_processing: str = 'splade_max',
         doc_processing: str = 'splade_max',
-        pooling_type="cls",
+        pooling_type="none",
         inbatch_loss=None,
         model_tied=True,
         use_pooler=False,
@@ -48,7 +48,7 @@ def from_pretrained(
         model_name_or_path: str = "bert-base-uncased",
         query_processing: str = 'splade_max',
         doc_processing: str = 'splade_max',
-        pooling_type="cls",
+        pooling_type="none",
         inbatch_loss=None,
         model_tied=True,
         use_pooler=False,