Models GEC spanish
Collection
24 items • Updated
How to use Ro551/mbart-large-50-GEC-spanish-LORA-cowsl2h 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-cowsl2h")How to use Ro551/mbart-large-50-GEC-spanish-LORA-cowsl2h with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Ro551/mbart-large-50-GEC-spanish-LORA-cowsl2h", device_map="auto")This model is a fine-tuned version of facebook/mbart-large-50 on an unknown 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 |
|---|---|---|---|---|
| 9.2131 | 1.0 | 388 | 8.1384 | 0.4301 |
| 8.1633 | 2.0 | 776 | 8.1093 | 0.5478 |
| 8.1815 | 3.0 | 1164 | 8.1496 | 0.3411 |
| 8.1746 | 4.0 | 1552 | 8.1436 | 0.3894 |
Base model
facebook/mbart-large-50
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Ro551/mbart-large-50-GEC-spanish-LORA-cowsl2h", device_map="auto")