Downloads · 30 days
26
1% of all-time downloads
unicamp-dl/mt5-base-en-msmarco
mt5-base-en-msmarco is a machine learning model from unicamp-dl. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
mT5-base-en-msmarco-v1 is a mT5-based model finetuned on English MS MARCO passage dataset. Further information about the dataset or the translation method can be found on our paper mMARCO: A Multilingual Version of th…
Downloads · 30 days
26
1% of all-time downloads
All-time downloads
2.5K
Public
Repo size
4.7 GB
Likes
0
Public
Click a slice to open those files.
.bin2.3 GB · 100%
From the Hugging Face model README
mT5-base-en-msmarco-v1 is a mT5-based model finetuned on English MS MARCO passage dataset. Further information about the dataset or the translation method can be found on our paper mMARCO: A Multilingual Version of the MS MARCO Passage Ranking Dataset and mMARCO repository.
from transformers import T5Tokenizer, MT5ForConditionalGeneration
model_name = 'unicamp-dl/mt5-base-en-msmarco'
tokenizer = T5Tokenizer.from_pretrained(model_name)
model = MT5ForConditionalGeneration.from_pretrained(model_name)
If you use mT5-base-en-msmarco, please cite:
@misc{bonifacio2021mmarco,
title={mMARCO: A Multilingual Version of the MS MARCO Passage Ranking Dataset},
author={Luiz Henrique Bonifacio and Vitor Jeronymo and Hugo Queiroz Abonizio and Israel Campiotti and Marzieh Fadaee and and Roberto Lotufo and Rodrigo Nogueira},
year={2021},
eprint={2108.13897},
archivePrefix={arXiv},
primaryClass={cs.CL}
}