bmd1905/vi-error-correction-v2
Viewer • Updated • 360k • 41 • 3
How to use tminhtri1910/bartpho-vietnamese-correction-bmd1905 with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("vinai/bartpho-syllable")
model = PeftModel.from_pretrained(base_model, "tminhtri1910/bartpho-vietnamese-correction-bmd1905")How to use tminhtri1910/bartpho-vietnamese-correction-bmd1905 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("tminhtri1910/bartpho-vietnamese-correction-bmd1905", dtype="auto")This model is a fine-tuned version of vinai/bartpho-syllable 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 | Sacrebleu |
|---|---|---|---|---|
| 0.5948 | 1.0 | 2657 | 0.4292 | 24.7435 |
| 0.5053 | 2.0 | 5314 | 0.3612 | 26.2000 |
| 0.4854 | 3.0 | 7971 | 0.3466 | 26.5168 |
Base model
vinai/bartpho-syllable