--- license: mit tags: - generated_from_trainer datasets: - samsum metrics: - rouge base_model: facebook/bart-large-xsum model-index: - name: bart-large-xsum-finetuned-samsum-v2 results: - task: type: text2text-generation name: Sequence-to-sequence Language Modeling dataset: name: samsum type: samsum config: samsum split: train args: samsum metrics: - type: rouge value: 54.1988 name: Rouge1 --- # bart-large-xsum-finetuned-samsum-v2 This model is a fine-tuned version of [facebook/bart-large-xsum](https://huggingface.co/facebook/bart-large-xsum) on the samsum dataset. It achieves the following results on the evaluation set: - Loss: 1.4078 - Rouge1: 54.1988 - Rouge2: 29.3478 - Rougel: 44.6308 - Rougelsum: 49.8443 - Gen Len: 29.0538 ## 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: 2e-05 - train_batch_size: 4 - eval_batch_size: 4 - seed: 42 - gradient_accumulation_steps: 2 - total_train_batch_size: 8 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - num_epochs: 1 - mixed_precision_training: Native AMP ### Training results | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len | |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:| | 1.3515 | 1.0 | 1841 | 1.4078 | 54.1988 | 29.3478 | 44.6308 | 49.8443 | 29.0538 | ### Framework versions - Transformers 4.25.1 - Pytorch 1.13.0+cu116 - Datasets 2.7.1 - Tokenizers 0.13.2