使用 AWS Trainium 微调 Transformers

在 AWS Trainium 上进行训练与在 Transformers 中一样简单

• 您需要将 Transformers 的 Trainer 类替换为 NeuronTrainer 类。

您可以在官方存储库中找到以下任务的多个示例

• 语言建模,
• 问答,
• 摘要,
• 文本分类,
• 翻译,
• 图像分类,
• 音频分类,
• 语音识别,
• 对比图像-文本训练.

如果您想逐步了解示例，请查看 AWS Trainium 和 Hugging Face Transformers 入门 指南。