distilbart-cnn-12-6-tuned-0104-0138

This model is a fine-tuned version of sshleifer/distilbart-cnn-12-6 on the broadfield-dev/stacked-summaries_xsum-2048-in dataset.

Training Details

• Task: SEQ_2_SEQ_LM
• Epochs: 2
• Learning Rate: 2e-05
• Gradient Accumulation Steps: 4

Entity Labels

['LABEL_0', 'LABEL_1', 'LABEL_2']

Usage

from transformers import pipeline

summarizer = pipeline("summarization", model="broadfield-dev/distilbart-cnn-12-6-tuned-0104-0138")
text = "Your long text here..."
print(summarizer(text))