如何读懂Miles启动脚本:4步拆解train_args组装与execute_train提交流程

发布时间:2026/9/17 8:34:13

如何读懂Miles启动脚本:4步拆解train_args组装与execute_train提交流程 如何读懂Miles启动脚本4步拆解train_args组装与execute_train提交流程【免费下载链接】milesMiles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.项目地址: https://gitcode.com/GitHub_Trending/miles1/milesMiles 是一个面向企业级场景的大模型LLM/VLM后训练强化学习框架从 slime 分叉并共同演进。它把「每个支持的模型 一个 Python 启动脚本」作为核心设计你只需运行一条命令如python scripts/run_qwen3_4b.py脚本就会自动组装完整的train.py命令行参数即train_args并通过execute_train启动 Ray 集群、把训练任务提交上去。本文带你 4 步读懂这条从「脚本选项」到「分布式训练」的完整链路无需阅读任何源码细节也能上手。第1步先建立全局认知——Miles启动脚本的4层分工一个启动脚本只是「配方」不是训练进程本身。整条链路可以拆成 4 层看懂这个表格你再看任何一个scripts/run_*.py都不会迷路层级位置职责启动脚本scripts/run_qwen3_4b.py 等scripts/run_*.py保存模型配方脚本选项 各功能块的训练参数模型定义scripts/models/按megatron_model_type提供 Megatron 架构参数命令工具miles/utils/external_utils/command_utils.pyexecute_train起 Ray、提交作业训练入口train.py / train_async.py真正的训练进程运行在 Ray 作业内部上图展示了脚本提交作业后 Ray 集群内的分工数据缓冲区把 prompt 交给自定义 rollout 生成逻辑SGLang router 将请求分发给多个 SGLang 推理引擎Megatron 训练器完成梯度更新后把新权重同步回推理侧——这正是train_args里那些--rollout-*、--sglang-*参数背后的系统。第2步读懂ScriptArgs——dataclass如何变成命令行选项打开 scripts/run_qwen3_4b.py第一个要看的类是ScriptArgs它继承自 ExecuteTrainConfig自带num_nodes、output_dir等所有启动器共享的选项脚本自身的选项model_name、hardware、train_backend、enable_eval……都是普通字段文件底部的U.dataclass_cli装饰器把每个字段自动暴露为两种方式--kebab-case命令行参数如--model-dir和MILES_SCRIPT_*前缀环境变量如MILES_SCRIPT_MODEL_DIR。优先级顺序很直白命令行 环境变量 字段默认值。环境变量这条通道是集群工具注入机器相关配置的「后门」你不需要改脚本一行代码。ScriptArgs.__post_init__还负责做两件推导自动检测硬件并换算每节点 GPU 数U.resolve_hardwareNUM_GPUS_OF_HARDWARE以及按模型名推导默认并行策略TP/PP/CP、序列并行开关、max_tokens_per_gpu等。第3步execute()解剖——train_args由哪几个参数块拼成execute(args)是train_args的组装车间。它按「一个关注点一个字符串块」的方式构建参数最后按固定顺序拼接train_args ckpt_args rollout_args optimizer_args grpo_args wandb参数 perf_args eval_args ci_args sglang_args train_backend_args misc_args true_on_policy_args args.extra_args各块的分工对照官方 Argument Groups 文档 一句话就能记住参数块管什么典型标志ckpt_args三个模型角色的路径--hf-checkpoint、--ref-load、--load、--saverollout_args数据入口与采样量--prompt-data、--rollout-batch-size、--n-samples-per-prompt、--rm-typegrpo_argsRL 目标函数--advantage-estimator grpo、--eps-clip、--use-kl-lossoptimizer_args优化器与学习率--optimizer adam、--lr 1e-6train_backend_args按后端分叉fsdp/megatronTP/PP/CP 并行度、重计算、动态 batchsglang_args推理引擎拓扑与透传--rollout-num-gpus-per-engine、--sglang-mem-fraction-staticmisc_argsGPU 布局与杂项--actor-num-nodes、--colocate、--dump-details有两个「隐藏块」值得知道模型架构参数不由脚本拼接。Megatron 无法从 HuggingFace checkpoint 推断全部结构所以execute_train会按megatron_model_type到 scripts/models/ 下按文件名解析对应脚本如 scripts/models/qwen3-4B.py 返回--num-layers 36 --hidden-size 2560 …并把这些架构参数插到train_args之前。wandb 参数是条件性的。U.get_default_wandb_args只在检测到WANDB_API_KEY时才返回--use-wandb系列参数——导出密钥即开启日志零脚本改动。第4步execute_train做了什么——从参数串到Ray作业组装完的train_args交给 execute_train()它依次执行 5 件事日志里都会带EXEC:前缀逐条打印相当于完整的执行记录清场杀掉残留的sglang、miles、redis进程ray stop --force停掉旧集群起集群ray start --head --num-gpus num_gpus_per_node启动新的 Ray head 节点执行提交前钩子若传了before_ray_job_submit例如 ssh_start_ray_workers 通过 ssh 把多机拉进集群此时运行构建运行时环境注入PYTHONUNBUFFERED、CUDA_DEVICE_MAX_CONNECTIONS1仅 Megatron 后端、NCCL_NVLS_ENABLE自动探测 NVLink、MASTER_ADDR、no_proxy以及把仓库根目录和--megatron-path拼进PYTHONPATH提交作业ray job submit -- python3 train.py 架构参数 train_args。两个环境变量可以跳过快照中的部分步骤MILES_SCRIPT_EXTERNAL_RAY1表示集群已存在跳过清场和ray startMILES_SCRIPT_ENABLE_RAY_SUBMIT0则只演练不提交适合排障。提交之后train.py的三段式训练循环作业进入 train.py 后流程一目了然parse_args()解析全部参数 →create_rollout_components拉起带 SGLang 引擎的 rollout 管理器 →create_training_models建 actor/critic → 先update_weights保证推理侧拿到训练权重 → 进入 rollout 循环采样 → 训练 → 周期性save/eval→ 回传权重。训练效果在曲线上长这样常见问题不改脚本如何覆盖配方方式做法适用场景追加--extra-args末尾追加的参数同名覆盖前面的快速试验--extra-args --lr 2e-6环境变量MILES_SCRIPT_MODEL_DIR/mnt/models python scripts/run_qwen3_4b.py集群工具注入机器相关路径直接编辑脚本改ScriptArgs或参数块想长期保留的改动更多细节可查官方文档 Launch Script 指南、Argument Groups 分组说明 与 CLI Reference。相关文件速查单模型启动脚本scripts/run_qwen3_4b.py、scripts/run_deepseek_v4.py模型架构参数目录scripts/models/每个megatron_model_type一个同名文件命令工具核心miles/utils/external_utils/command_utils.py训练入口train.py、train_async.py【免费下载链接】milesMiles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.项目地址: https://gitcode.com/GitHub_Trending/miles1/miles创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
延伸阅读

更多相关文章

2026/9/17 8:34:13

数学建模智能体实战:三角色协作与数值校验闭环

几个月前,我一直在琢磨一个问题:现在的语言模型写文档、写代码已经挺像样了,可一旦遇到“给一堆约束条件求最优解”这类正经的数学建模问题,它们就很容易翻车。不是因为模型不够聪明,而是因为数学建模本身就不该靠“想…

2026/9/17 8:34:13

基于PPO算法的无人机姿态控制:PyTorch与AirSim实战解析

简介:无人机自主导航中的姿态控制调参,是强化学习从仿真走向落地的重要难题。一份40页的PDF文档对此做了系统梳理,聚焦PPO算法在PyTorch与AirSim仿真平台上的完整实现与参数调节技巧。它面向无人机开发者、强化学习研究者和PyTorch使用者&…

2026/9/17 9:34:22

搜广推排序模型演进:PageRank到LambdaMART的工程实战

1. 这不是“算法科普”,而是搜广推工程师每天要面对的真实战场你点开淘宝搜“无线耳机”,第一页出现的为什么是那几家店?不是因为它们交了更多广告费,也不是因为平台偏心——而是背后一整套精密运转的排序模型在实时决策。PageRan…

2026/9/17 9:34:22

AI编程工具深度评测:六款主流产品实测与选型避坑指南

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/9/17 9:34:22

预测赢不等于交易赢:Transformer、LSTM、TCN、MLP高频实测

模型把涨跌预测准确率刷到 62%,回测曲线漂亮得跟画出来似的,结果一挂到模拟盘上,手续费加滑点直接把这个优势吃干抹净——这种场面我这些年见过太多次了。这次我干脆把家里压箱底的四种深度学习架构全搬出来,做了一次横向评测&…

2026/9/17 9:34:22

量子计算开发者的心理健康挑战与应对策略

1. 量子计算时代开发者面临的心理健康挑战量子计算技术的快速发展正在重塑整个软件开发行业。作为站在技术前沿的量子开发者,我们不仅要应对传统软件开发中的压力,还需要面对量子领域特有的认知负荷和工作模式转变。2026年的量子开发环境将比现在更加复杂…

2026/9/17 9:34:22

局域网IP排障:DOS命令查IP地址与实用技巧

简介:这是一份聚焦局域网IP地址查询与网络连通性检测的实用PDF手册,适合网络管理员、IT运维人员及刚接触DOS命令的入门用户。文档围绕Ping和Netstat两大核心命令展开。Ping部分从基础入手,介绍了ping 127.0.0.1验证本机协议栈、ping本机IP确认…

2026/9/16 12:52:37

拯救者Y7000黑屏故障排查与维修实战指南

1. 项目概述:一台黑屏的拯救者Y7000,到底卡在哪一步? 联想拯救者Y7000系列笔记本,从2018年第一代搭载i5-8300H开始,到后来的i7-9750H、i7-10750H、i5-11400H,再到2023年款的R7-7840HS,它始终是学…

2026/9/17 0:03:13

WiFi密码安全测试:从原理到实战的字典暴力破解指南

1. 写在前面:我为什么要研究WiFi密码这件事先交代一下背景。我身边有不少朋友,家里的WiFi密码常年是"12345678"或者"88888888",问就是"好记"。直到有一次,隔壁邻居蹭网蹭到我家路由器后台都进不去&…

2026/9/17 0:03:13

redis-py服务控制与监控函数实战:从ping到slowlog的巡检指南

我用 redis-py 写了快五年的业务代码,坦白说,真正让我觉得这个客户端“像一个成熟工具箱”的,不是 get/set 那套基本操作,而是它那批专门做服务控制与状态监控的辅助函数。日常开发里,大家把redis.Redis(host..., deco…

2026/9/17 0:03:13

SpringBoot+Vue3实现中小企业设备管理系统开发实践

1. 项目概述与核心价值中小企业设备管理系统是制造业、服务业等领域的基础信息化工具。传统设备管理往往依赖Excel表格或纸质记录,存在数据孤岛、流程混乱、维护成本高等痛点。这套基于Java SpringBootVue3MyBatis的技术方案,通过前后端分离架构实现了设…

2026/9/16 22:55:57

USB Type-C PCB布局分区设计:电源、高速信号与PD协议全攻略

做硬件这行,Type-C接口算是典型的“看着简单,做起来全坑”的东西。光引脚就24个,高低速信号、电源、控制线全部塞在一个小小的连接器里,如果PCB布局不做规划,打样回来基本就是“插上没反应”、“高速掉线”、“静电一打…

2026/9/16 22:56:09

系统编程学习原型如何补齐稳定性边界

系统编程学习原型如何补齐稳定性边界预算有限时&#xff0c;我先优化明显多余的复制&#xff0c;而不是猜测性地换容器。用借用传递只读数据通常就能减少分配&#xff1a; fn parse(line: &str) -> Result<Item, Error> { /* ... */ }用基准确认热点确实在分配&am…

2026/9/16 22:56:16

雨花区哪家财务公司代理记账比较好?

在雨花区&#xff0c;企业处理财税事务常常面临诸多挑战&#xff0c;选择一家靠谱的财务公司至关重要。湖南巨勤财务管理咨询有限公司就是本地正规实体财税服务机构&#xff0c;深耕本地工商财税行业多年&#xff0c;熟悉当地工商局、税务局最新政策与申报流程。主营公司注册、…

还想了解更多?直接咨询顾问

免费诊断 + 免费方案 + 透明报价。

全国咨询热线400-8866-253
免费获取方案
咨询二维码