Bart Base Xsum

🚀 bart-base-xsum

bart-base-xsum 是基於 facebook/bart-base 在 xsum 數據集上微調的模型，可用於文本摘要任務，在評估集上取得了良好的效果。

🚀 快速開始

此模型是 facebook/bart-base 在 xsum 數據集上的微調版本。它在評估集上取得了以下結果：

• 損失率（Loss）：0.8051
• ROUGE-1（R1）：0.5643
• ROUGE-2（R2）：0.3017
• ROUGE-L（Rl）：0.5427
• ROUGE-LSUM（Rlsum）：0.5427

模型訓練使用了以下倉庫中的腳本：https://github.com/MorenoLaQuatra/transformers-tasks-templates

📚 詳細文檔

訓練和評估數據

更多信息待補充。

模型描述

更多信息待補充。

預期用途和限制

更多信息待補充。

🔧 技術細節

訓練過程

訓練超參數

訓練期間使用了以下超參數：

• 學習率（learning_rate）：5e-05
• 訓練批次大小（train_batch_size）：32
• 評估批次大小（eval_batch_size）：32
• 隨機種子（seed）：42
• 優化器（optimizer）：Adam，β1 = 0.9，β2 = 0.999，ε = 1e-08
• 學習率調度器類型（lr_scheduler_type）：線性
• 學習率調度器熱身步數（lr_scheduler_warmup_steps）：500
• 訓練輪數（num_epochs）：10

訓練結果

框架版本

• Transformers：4.22.1
• Pytorch：1.11.0+cu113
• Datasets：2.0.0
• Tokenizers：0.11.6

📄 許可證

本項目採用 Apache-2.0 許可證。

模型指標詳情