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