Models GEC spanish
Collection
24 items • Updated
How to use Ro551/mt5-small-GEC-spanish-LORA-cowsl2h with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-small")
model = PeftModel.from_pretrained(base_model, "Ro551/mt5-small-GEC-spanish-LORA-cowsl2h")How to use Ro551/mt5-small-GEC-spanish-LORA-cowsl2h with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Ro551/mt5-small-GEC-spanish-LORA-cowsl2h", device_map="auto")This model is a fine-tuned version of google/mt5-small 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 |
|---|---|---|---|---|
| 2.8700 | 1.0 | 388 | 1.0861 | 0.0 |
| 0.4864 | 2.0 | 776 | 0.1614 | 0.2769 |
| 0.3819 | 3.0 | 1164 | 0.1458 | 0.3180 |
| 0.3567 | 4.0 | 1552 | 0.1365 | 0.3258 |
Base model
google/mt5-small