RL-Factory 的项目扩展与二次开发
2025-05-26 00:28:13作者:尤辰城Agatha
1. 项目的基础介绍
RL-Factory 是一个为 Agentic Learning 设计的简单且高效的 RL(Reinforcement Learning,强化学习)后训练框架。它将环境与 RL 后训练解耦,允许用户仅通过工具配置和奖励函数即可进行训练,同时支持异步工具调用,使得 RL 后训练速度提高2倍。目前版本原生支持一键 DeepSearch 训练,并具备多轮工具调用、模型评判奖励以及多模型训练(包括 Qwen3)等功能。
2. 项目的核心功能
- 简单易设计的奖励函数:通过规则、模型评判甚至工具来计算奖励,满足各种奖励函数的要求。
- 无缝工具设置:仅需提供 MCP 工具和自定义工具的配置文件,即可集成到 RL 学习中。
- 多代理扩展:将代理转换为 MCP 格式,以便轻松实现多代理交互。
- 高效的工具调用:通过批处理和异步并行工具调用,提高在线 RL 训练的效率。
- 高效的奖励计算:分布式部署 LRM(如 QwQ-32B)进行高效的模型评判,并使用异步并行加速奖励计算。
3. 项目使用了哪些框架或库?
项目主要使用了以下框架和库:
- Python:基础编程语言。
- Cuda:用于加速 GPU 计算。
- vllm:支持 Qwen3 模型。
- accelerate、bitsandbytes、datasets、deepspeed、einops、flash-attn、isort、jsonlines、loralib、optimum、packaging、peft、pynvml、ray、tensorboard、torch、torchmetrics、tqdm、transformers、transformers_stream_generator、wandb:这些库用于支持项目中的各种功能,包括数据处理、模型训练等。
4. 项目的代码目录及介绍
项目的代码目录如下:
- assets:存放项目相关资源。
- docker:存放 Docker 相关配置。
- docs:项目文档。
- envs:环境配置相关代码。
- examples:示例代码和项目实例。
- generator:生成器相关代码。
- patches:补丁文件。
- rag_server:rag_server 相关代码。
- recipe:食谱或配置文件。
- scripts:脚本文件。
- tests:测试代码。
- verl:项目核心代码。
- webui:WebUI 相关代码。
- workspace/:工作空间目录。
- tools/:工具目录。
5. 对项目进行扩展或者二次开发的方向
- 新增模型支持:扩展框架以支持更多类型的强化学习模型,提高框架的通用性。
- 增加工具集:集成更多工具,提供更丰富的工具库,以适应不同领域和场景的需求。
- 优化训练算法:改进现有训练算法,提高训练效率和质量。
- WebUI 功能完善:完善 WebUI 功能,提供更友好的用户界面和交互体验。
- 多语言支持:扩展框架以支持其他编程语言,拓宽用户群体。
- 社区合作:鼓励和吸引更多开发者参与项目,共同推动项目的发展和完善。
登录后查看全文
热门项目推荐
相关项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0186
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0111
Step-3.7-FlashStep-3.7-Flash是一个拥有 1980 亿参数的稀疏混合专家(MoE)视觉语言模型,由 1960 亿参数的语言主干网络和 18 亿参数的视觉编码器组合而成,具备原生图像理解能力。Python00
JoyAI-EchoJoyAI-Echo,这是一个独立的、仅用于推理的版本,旨在实现分钟级多镜头音视频生成。它采用了经过蒸馏的DMD生成器、配对的跨模态记忆以及故事级别的一致性。其性能的核心在于,一个跨模态视听记忆库能够在长达五分钟的视频中保持角色外观和语音音色的一致性。同时,一个训练后处理流程将基于记忆的强化学习与分布匹配蒸馏相结合,实现了7.5倍的速度提升,显著增强了视觉质量和对齐效果。00
omega-aiOmega-AI:基于java打造的深度学习框架,帮助你快速搭建神经网络,实现模型推理与训练,引擎支持自动求导,多线程与GPU运算,GPU支持CUDA,CUDNN。Java03
llm-universe本项目是一个面向小白开发者的大模型应用开发教程,在线阅读地址:https://datawhalechina.github.io/llm-universe/Jupyter Notebook08
热门内容推荐
最新内容推荐
项目优选
收起
暂无描述
Dockerfile
759
4.94 K
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
853
1.91 K
deepin linux kernel
C
32
16
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
673
1.31 K
Ascend Extension for PyTorch
Python
716
866
Claude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed.
Get Started
Rust
1.76 K
185
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
454
436
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.06 K
1.09 K
CANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体,本仓库为其提供可复用的 Skills 模块。
Python
990
598
暂无简介
Dart
1 K
259