Models GEC spanish
Collection
24 items โข Updated
How to use Ro551/mbart-large-50-GEC-spanish-LORA-merged with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50")
model = PeftModel.from_pretrained(base_model, "Ro551/mbart-large-50-GEC-spanish-LORA-merged")How to use Ro551/mbart-large-50-GEC-spanish-LORA-merged with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Ro551/mbart-large-50-GEC-spanish-LORA-merged", device_map="auto")This model is a fine-tuned version of facebook/mbart-large-50 on the None dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Gleu |
|---|---|---|---|---|
| 5.4667 | 1.0 | 11213 | 6.8943 | 0.8234 |
| 5.4257 | 2.0 | 22426 | 6.8877 | 0.8475 |
Base model
facebook/mbart-large-50