somosnlp-hackathon-2022/neutral-es
Viewer • Updated • 3.61k • 53 • 7
How to use carmengoar/mbart-neutralization with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("carmengoar/mbart-neutralization")
model = AutoModelForSeq2SeqLM.from_pretrained("carmengoar/mbart-neutralization")This model is a fine-tuned version of t5-small on an unknown dataset. It achieves the following results on the evaluation set:
This model has been created as part of an activity for the course "Traducción Automática", at the Máster de Procesamiento del Lenguaje y Aplicaciones de la Inteligencia Artificial a la Lingüística (Universidad de La Rioja).
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
|---|---|---|---|---|---|
| No log | 1.0 | 440 | 0.2031 | 53.022 | 18.5417 |
| 0.2886 | 2.0 | 880 | 0.1563 | 53.022 | 18.5417 |
Base model
google-t5/t5-small