MonoT5¶
The MonoT5
module is a reranker that uses the MonoT5 model. This model rerank passages based on their relevance to a
given query.
Module Parameters¶
(Optional)
model_name
:Requiring the specification of a model_name. The model used must be defined in a provided model dictionary, allowing for flexible integration of different monoT5 variants.
default is
castorini/monot5-3b-msmarco-10k
(Optional)
batch
Specify the batch size of the query to the TART model.
default is 64.
Example config.yaml¶
modules:
- module_type: monot5
Supported Model Names¶
Model Name |
---|
castorini/monot5-base-msmarco |
castorini/monot5-base-msmarco-10k |
castorini/monot5-large-msmarco |
castorini/monot5-large-msmarco-10k |
castorini/monot5-base-med-msmarco |
castorini/monot5-3b-med-msmarco |
castorini/monot5-3b-msmarco-10k |
unicamp-dl/mt5-base-en-msmarco |
unicamp-dl/ptt5-base-pt-msmarco-10k-v2 |
unicamp-dl/ptt5-base-pt-msmarco-100k-v2 |
unicamp-dl/ptt5-base-en-pt-msmarco-100k-v2 |
unicamp-dl/mt5-base-en-pt-msmarco-v2 |
unicamp-dl/mt5-base-mmarco-v2 |
unicamp-dl/mt5-base-en-pt-msmarco-v1 |
unicamp-dl/mt5-base-mmarco-v1 |
unicamp-dl/ptt5-base-pt-msmarco-10k-v1 |
unicamp-dl/ptt5-base-pt-msmarco-100k-v1 |
unicamp-dl/ptt5-base-en-pt-msmarco-10k-v1 |
unicamp-dl/mt5-3B-mmarco-en-pt |
unicamp-dl/mt5-13b-mmarco-100k |