EdinburghNLP/xsum
Viewer โข Updated โข 227k โข 16.4k โข 157
How to use oslusarczyk/bbc_model_output with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("oslusarczyk/bbc_model_output")
model = AutoModelForSeq2SeqLM.from_pretrained("oslusarczyk/bbc_model_output", device_map="auto")This model is a fine-tuned version of google-t5/t5-small on the xsum 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 | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|---|---|---|---|---|---|---|---|---|
| No log | 1.0 | 50 | 2.8602 | 0.207 | 0.0363 | 0.1593 | 0.1595 | 19.235 |
| No log | 2.0 | 100 | 2.8428 | 0.2209 | 0.0377 | 0.1677 | 0.1677 | 19.17 |
Base model
google-t5/t5-small