Distilbart Cnn 12 6 Samsum

🚀 distilbart-cnn-12-6-samsum

该模型使用亚马逊SageMaker和全新的Hugging Face深度学习容器进行训练。它可用于文本摘要任务，能帮助用户快速提取文本的关键信息。

🚀 快速开始

此模型借助亚马逊SageMaker和Hugging Face深度学习容器进行训练。如需更多信息，可查看以下链接：

• 🤗 Transformers文档：亚马逊SageMaker
• 示例笔记本
• 亚马逊SageMaker的Hugging Face文档
• Python SDK SageMaker的Hugging Face文档
• 深度学习容器

✨ 主要特性

超参数

{
    "dataset_name": "samsum",
    "do_eval": true,
    "do_train": true,
    "fp16": true,
    "learning_rate": 5e-05,
    "model_name_or_path": "sshleifer/distilbart-cnn-12-6",
    "num_train_epochs": 3,
    "output_dir": "/opt/ml/model",
    "per_device_eval_batch_size": 8,
    "per_device_train_batch_size": 8,
    "seed": 7
}

📚 详细文档

训练结果

评估结果

💻 使用示例

基础用法

from transformers import pipeline
summarizer = pipeline("summarization", model="philschmid/distilbart-cnn-12-6-samsum")

conversation = '''Jeff: Can I train a 🤗 Transformers model on Amazon SageMaker? 
Philipp: Sure you can use the new Hugging Face Deep Learning Container. 
Jeff: ok.
Jeff: and how can I get started? 
Jeff: where can I find documentation? 
Philipp: ok, ok you can find everything here. https://huggingface.co/blog/the-partnership-amazon-sagemaker-and-hugging-face                                           
'''
nlp(conversation)

📄 许可证

本项目采用Apache-2.0许可证。