C✳CODEZ.WIN
🌐 简体中文

Skill 多语言阅读指南

trl-training用途、用法与原始来源

使用 TRL(Transformers 强化学习)对 LLM 进行后训练——包括 SFT、DPO、GRPO、KTO 和奖励模型训练。在编写或调试使用 TRL Python API 或 trl CLI 的训练代码时可以使用。

原文采集时间:

来源仓库
huggingface/skills
SKILL.md
skills/trl-training/SKILL.md
仓库 Star
11,241
仓库最近更新
2026/10/30
能力分类
开发
最近采集时间

许可:Apache-2.0

流程片段用于帮助阅读上游说明。本页没有执行第三方 Skill,不属于实测评测。

下面的用途和操作片段来自上游文档,由本站整理并翻译,和另行标注的 Codez 编辑分析分开展示。

这个 Skill 能做什么

使用 TRL(Transformers 强化学习)对 LLM 进行后训练——包括 SFT、DPO、GRPO、KTO 和奖励模型训练。在编写或调试使用 TRL Python API 或 trl CLI 的训练代码时可以使用。

如何使用:上游操作流程

TRL

每种方法都配对*Trainer一个班级*Config数据类。配置扩展transformers.TrainingArguments因此,它的所有参数在任何训练器配置中都有效。还有许多其他训练器(OnlineDPO、ORPO、CPO、GKD 等)也存在于此。trl.experimentalAPI不稳定:https://huggingface.co/docs/trl/experimental_overview

原文片段 · L10 ↗

数据集格式

对话式:{"messages": [{"role": ..., "content": ...}]}(语言建模)或{"prompt": [...], "completion": [...]}聊天模板会自动应用——切勿手动应用。允许添加额外列;GRPO 会将这些列转发给奖励功能。参考:https://huggingface.co/docs/trl/dataset_formats

原文片段 · L39 ↗

GRPO:在线RL

奖励函数通过关键字参数调用。prompts,completions,completion_ids,trainer_state加上每个额外的数据集列——接受kwargs对于那些你忽略的人。返回list[float]每次完成任务即可获得一份奖励。利用对话数据,completions是一个消息列表的列表,而不是字符串。生成批次是per_device_train_batch_size × num_processes × steps_per_generation(或设置)generation_batch_size直接)且必须能被……整除num_generations(默认8生成通常是瓶颈——启用 vLLMuse_vllm=True:vllm_mode="colocate"共享训练 GPU(大小为vllm_gpu_memory_utilization);vllm_mode="server"使用单独的trl vllm-serve --model 。

原文片段 · L55 ↗
原文片段 · python ↗
from datasets import load_dataset
from trl import SFTConfig, SFTTrainer

trainer = SFTTrainer(
    model="Qwen/Qwen2.5-0.5B",  # model ID or a PreTrainedModel instance
    args=SFTConfig(output_dir="Qwen2.5-0.5B-SFT"),
    train_dataset=load_dataset("trl-lib/Capybara", split="train"),
)
trainer.train()

环境与依赖

本次未从该文件选出独立的环境要求片段,不代表无需依赖。请继续核对仓库的安装说明和引用文件。

限制与结果检查

本次未从该文件选出独立的限制片段,不代表适用于所有输入或环境。请阅读完整原文,并自行验证实际结果。

开始使用前

  1. 先核对支持的 Agent 宿主和仓库安装说明。保留完整 Skill 目录及其引用的脚本、资源和文档,只复制 SKILL.md 可能漏掉依赖。
  2. 向 Agent 提供实际输入、期望输出和验收标准。调用方式与可用工具取决于宿主环境,本站不假定存在通用的安装命令。
  3. 先在临时项目或输入副本上试用。涉及修改文件、远程服务或计费资源的操作,应逐项核对后再授权。
  4. 按照原文检查项和自己的需求验证输出。成功触发或仓库 Star 较高,都不能证明结果正确。

Skill 文件与目录格式规范 ↗

继续阅读原始文档

翻译与介绍范围

本页从固定版本的上游 SKILL.md 整理用途与操作片段,译文使用辅助翻译。它不是完整手册,也不代表经过兼容性实测;使用前请核对原文及其引用文件。

介绍依据下方列出的固定版本;当前仓库动态和目录采集时间单独展示。

固定版本来源
阅读原文 ↗
原文采集时间
来源提交版本
fabc634a893f62ad5c2039830397efa35fa62715
原文 SHA-256
42c855bdc6171390abc7413cdb19bd4fd8dddfb31094d519a874427de08ddc2a
返回完整 Skill 目录