TRL documentation

TRL - 基于 Transformers 的强化学习

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.17.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

TRL - 基于 Transformers 的强化学习

TRL 是一个全栈库,我们提供了一系列工具,用于使用监督微调(SFT)、组相对策略优化(GRPO)、直接偏好优化(DPO)、奖励建模等方法训练 Transformer 语言模型。 该库与 🤗 transformers 集成。

🎉 新功能

🔥 Mistral Large 4:在单个 8 卡 B300 节点上,基于其 FP8 权重,使用 SFTTrainer 对 Mistral Large 4 进行微调。请参阅 LoRA 示例。

🚀 在 AMD GPU 上进行训练:TRL 的训练器现在可以通过 PyTorch 的 ROCm 构建版本在 AMD Instinct GPU 上运行。

🔥 Beam:使用 SFTTrainer 在单节点 8 张 B300 上对 Beam 进行微调。请参阅 LoRA 示例。

🔥 Muse Spark Open:使用 SFTTrainer 对 Muse Spark Open 进行微调,并在加载时将其分片到各 GPU 上。请参阅 8 节点 LoRA 示例。

🚀 在 AWS Trainium 上进行训练:TRL 的训练器现在可以通过 TorchNeuron(PyTorch 的原生 Neuron 后端)在 AWS Trainium 上运行。

分类

以下是按成熟度和方法类型整理的 TRL 训练器概览。

在线方法

奖励建模

实验性

您还可以在 TRL Hugging Face 组织 中探索与 TRL 相关的模型、数据集和演示。

学习

在 🤗 smol course 中,学习使用 TRL 及其他库进行后训练。

目录

文档分为以下几个部分:

  • 入门指南:安装和快速入门指南。
  • 概念指南:数据集格式、训练常见问题解答以及日志解读。
  • 操作指南:减少内存占用、加快训练速度、分布式训练等。
  • 集成:DeepSpeed、PEFT 等
  • 示例:示例概览、社区教程等。
  • API:训练器、工具等。

博客文章

演讲

Update on GitHub