EdinburghNLP/xsum
Viewer โข Updated โข 227k โข 16.4k โข 157
How to use phdreg/t5-small-finetuned-xsum with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("phdreg/t5-small-finetuned-xsum")
model = AutoModelForSeq2SeqLM.from_pretrained("phdreg/t5-small-finetuned-xsum", device_map="auto")This model is a fine-tuned version of t5-small on the xsum dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|---|---|---|---|---|---|---|---|---|
| No log | 1.0 | 125 | 3.0093 | 17.8475 | 2.6854 | 14.028 | 14.2242 | 18.889 |
Base model
google-t5/t5-small