diff --git a/rankers/modelling/sparse.py b/rankers/modelling/sparse.py index 45081dc..6805376 100644 --- 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,