lhbonifacio's picture
Initial commit
2b989c7
|
raw
history blame
1.3 kB
metadata
language: pt
license: mit
tags:
  - msmarco
  - t5
  - pytorch
  - tensorflow
  - pt
  - pt-br
datasets:
  - msmarco
widget:
  - text: Texto de exemplo em português
inference: false

mt5-base-en-pt-msmarco Reranker finetuned on Multi MS MARCO

Introduction

mT5-base-en-pt-msmarco is a mT5-based model finetuned on a bilingual version of MS MARCO passage dataset. This bilingual dataset version is formed by the original MS MARCO dataset (in English) and a Portuguese translated version. Further information about the dataset or the translation method can be found on our Cross-Lingual repository.

Usage


from transformers import T5Tokenizer, MT5ForConditionalGeneration

model_name = 'unicamp-dl/mt5-base-en-pt-msmarco'
tokenizer  = T5Tokenizer.from_pretrained(model_name)
model      = MT5ForConditionalGeneration.from_pretrained(model_name)

Citation

If you use mt5-base-en-pt-msmarco, please cite:

@article{rosa2021cost,
  title={A cost-benefit analysis of cross-lingual transfer methods},
  author={Rosa, Guilherme Moraes and Bonifacio, Luiz Henrique and de Souza, Leandro Rodrigues and Lotufo, Roberto and Nogueira, Rodrigo},
  journal={arXiv preprint arXiv:2105.06813},
  year={2021}
}