Downloads · 30 days
14
1% of all-time downloads
bart-large-cnn-finetuned-multi-news
nikhedward/bart-large-cnn-finetuned-multi-news
bart-large-cnn-finetuned-multi-news is a machine learning model from nikhedward. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
should probably proofread and complete it, then remove this comment. --
All-time downloads
2K
Public
Repo size
4.9 GB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin1.6 GB · 100%
At a glance
- Library
- transformers
- License
- mit
- Model type
- bart
- Access
- Public
- Created
- Apr 13, 2022
- Updated
- Apr 29, 2022
- SHA
- 6c150c04
Datasets
Model card
From the Hugging Face model README
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
bart-large-cnn-finetuned-multi-news
This model is a fine-tuned version of facebook/bart-large-cnn on the multi_news dataset. It achieves the following results on the evaluation set:
- Loss: 2.0950
- Rouge1: 42.0423
- Rouge2: 14.8812
- Rougel: 23.3412
- Rougelsum: 36.2613
Model description
bart-large-cnn fine tuned on sample of multi-news dataset
Intended uses & limitations
The intended use of the model is for downstream summarization tasks but it's limited to input text 1024 words. Any text longer than that would be truncated.
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 4
- eval_batch_size: 4
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 16
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 1
Training results
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|---|---|---|---|---|---|---|---|
| 2.2037 | 1.0 | 750 | 2.0950 | 42.0423 | 14.8812 | 23.3412 | 36.2613 |
Framework versions
- Transformers 4.18.0
- Pytorch 1.10.0+cu111
- Datasets 2.0.0
- Tokenizers 0.11.6
- Library
- transformers
- Type
- bart
- License
- mit
- Created
- Apr 13, 2022
- Updated
- Apr 29, 2022