跳到正文
原文
Hugging Face Blog·· 2023-08-08精选AI 评分62

用 DPO 微调 Llama 2:TRL 新增 DPOTrainer 并给出完整训练代码

Fine-tune Llama 2 with DPO

AI 导读

Hugging Face 在 TRL 库中上线 Direct Preference Optimization(DPO)训练器,用二元交叉熵损失直接优化偏好数据,省去 RLHF 中的奖励模型训练和 RL 优化两步。

推荐理由

TRL 里新增 DPO 训练器,附 Llama 2 7B 的 QLoRA 微调完整代码,想省掉奖励模型和 RL 环节的可以直接照跑。

来源:Hugging Face Blog · huggingface.co