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

发布时间:2026/11/8 15:31:48

如何读懂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/11/8 22:28:28

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

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

2026/10/31 23:45:03

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

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

2026/11/8 22:27:39

巴南花溪考电工证网上查询报名费和考试全攻略

巴南花溪考电工证网上查询报名费和考试全攻略 工地太忙根本没时间复习考试,你是不是也经常在工地上被各种事务压得喘不过气?一边要赶工期,一边还得为电工证考试发愁,这事儿真不是一两句话能说清的。尤其是 巴南花溪考电工证…

2026/11/8 22:27:39

天山区电工作业真实经历:怕被中介坑?看看我怎么避雷的

天山区电工作业真实经历:怕被中介坑?看看我怎么避雷的 我就是天山区的一个普通电工,去年在报考低压电工证的时候,真的被几个“培训机构”坑得不轻。他们打着“包过”的旗号,收取高额费用,最后连报名都没报上。后来我通过朋友推荐,才知道天山区电工作业…

2026/11/8 22:27:39

钦州电工证培训难吗?网上查询最新流程和避坑指南

钦州电工证培训难吗?网上查询最新流程和避坑指南 之前考的证在外省能不能转过来?如果你是新乡房建工程从业者,这个问题你肯定遇到过。其实,很多电工在跨省工作时,都会遇到证书能否异地转换的困惑。别急,下面我们就来聊聊钦州电工证培训难吗,以及如何通…

2026/11/8 22:27:39

深圳高级电工证样本真实经历:证书过期了不知道怎么复审延期

深圳高级电工证样本真实经历:证书过期了不知道怎么复审延期 证书过期了不知道怎么复审延期,这事儿我真经历过,也帮过不少朋友解决过。前两天一个在许昌的应届工程类毕业生找我,说他的深圳高级电工证过期了,但不知道该怎么办,生怕影响工作和项目进展。这…

2026/11/8 22:27:39

电工证免考骗局频现,考前押题怎么找才靠谱

电工证免考骗局频现,考前押题怎么找才靠谱 实操考试心里没底怕挂科,特别是周口的市政公用工程从业者,很多人在报考电工证时,都被“免考骗局”搞得晕头转向。你可能看到一些培训机构打着“不用考试,直接拿证”的旗号,诱惑你交钱,结果一查,发现根本不是…

2026/11/8 22:22:38

为啥查不到电工证了呢官方回应来了!新乡人别慌,手把手教你应对

为啥查不到电工证了呢官方回应来了!新乡人别慌,手把手教你应对 急需特种作业证才能上岗找工作,偏偏查不到电工证信息,心里急得像热锅上的蚂蚁。最近有不少新乡朋友反馈,明明已经考了低压电工证,但去官网查不到信息,甚至有的证书被系统标记为“已注销”…

2026/11/8 0:00:27

电工证丢失还能补吗?千万别踩坑,丢了证等于失业

电工证丢失还能补吗?千万别踩坑,丢了证等于失业 你正在找工作,但手里的电工证突然丢了,这下怎么办?别急,先别慌, 电工证丢失还能补吗 ?答案是肯定的,但 千万别踩坑 ,补办流程如果不了解,可能会耽误你上岗、甚至影响到你的收入。…

2026/11/8 0:00:27

电工证丢失还能补吗?千万别踩坑,丢了证等于失业

电工证丢失还能补吗?千万别踩坑,丢了证等于失业 你正在找工作,但手里的电工证突然丢了,这下怎么办?别急,先别慌, 电工证丢失还能补吗 ?答案是肯定的,但 千万别踩坑 ,补办流程如果不了解,可能会耽误你上岗、甚至影响到你的收入。…

2026/11/7 14:54:47

面向复杂光照与角度的宠物面部识别鲁棒性增强策略

在智慧养宠时代,基于视觉的宠物身份识别技术正逐步从实验室走向家庭应用。然而,与配合度高的人类不同,宠物是典型的“非合作生物识别对象”——它们不会主动配合摄像头调整姿态,更无法控制光照环境。实际应用中,我们常…

2026/11/7 14:50:12

SSM电子书店源码全解析:结构、原理、部署与排错实战

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

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

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

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