--- license: apache-2.0 base_model: facebook/bart-base tags: - generated_from_trainer metrics: - rouge model-index: - name: bart-base-finetuned-CNN-DailyNews results: [] --- # bart-base-finetuned-CNN-DailyNews This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on the None dataset. It achieves the following results on the evaluation set: - Loss: 1.8693 - Rouge1: 0.1878 - Rouge2: 0.1058 - Rougel: 0.1653 - Rougelsum: 0.1752 ## Model description More information needed ## Intended uses & limitations More information needed ## Training and evaluation data More information needed ## Training procedure ### Training hyperparameters The following hyperparameters were used during training: - learning_rate: 5.6e-05 - train_batch_size: 8 - eval_batch_size: 8 - seed: 42 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - num_epochs: 8 ### Training results | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:| | 2.6219 | 1.0 | 63 | 1.9905 | 0.1791 | 0.1034 | 0.1598 | 0.169 | | 2.0525 | 2.0 | 126 | 1.8810 | 0.18 | 0.1038 | 0.1616 | 0.1701 | | 1.7989 | 3.0 | 189 | 1.8439 | 0.189 | 0.1064 | 0.1679 | 0.176 | | 1.6421 | 4.0 | 252 | 1.8411 | 0.1867 | 0.1066 | 0.1648 | 0.1739 | | 1.4826 | 5.0 | 315 | 1.8437 | 0.1862 | 0.107 | 0.1657 | 0.1748 | | 1.3858 | 6.0 | 378 | 1.8613 | 0.1881 | 0.1085 | 0.1671 | 0.1749 | | 1.2879 | 7.0 | 441 | 1.8674 | 0.1835 | 0.1021 | 0.1621 | 0.171 | | 1.2539 | 8.0 | 504 | 1.8693 | 0.1878 | 0.1058 | 0.1653 | 0.1752 | ### Framework versions - Transformers 4.35.2 - Pytorch 2.1.0+cu118 - Datasets 2.15.0 - Tokenizers 0.15.0