facebook/covost2
Updated โข 462 โข 52
How to use JohnLei/t5-small-translation_v3 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("JohnLei/t5-small-translation_v3")
model = AutoModelForSeq2SeqLM.from_pretrained("JohnLei/t5-small-translation_v3", device_map="auto")This model is a fine-tuned version of google-t5/t5-small on the covost2 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 | Bleu | Gen Len |
|---|---|---|---|---|---|
| 4.539 | 1.0 | 746 | 4.1366 | 0.0502 | 20.0 |
| 4.3657 | 2.0 | 1492 | 4.0781 | 0.0826 | 19.9982 |
| 4.2651 | 3.0 | 2238 | 4.0510 | 0.0589 | 19.9998 |
| 4.2638 | 4.0 | 2984 | 4.0376 | 0.1365 | 20.0 |
| 4.227 | 5.0 | 3730 | 4.0331 | 0.1366 | 20.0 |
Base model
google-t5/t5-small