HyperParallel 模型训练
HyperParallel 模型训练提供由 YAML 配置驱动的文本与多模态 Trainer。parse_training_args() 将 YAML 和 CLI dotted overrides 解析为 TrainerConfig,TextTrainer 或 VLMTrainer 再根据配置构建训练组件并执行训练。分布式张量、模型分片和集合通信等底层能力由 HyperParallel Core 提供。
使用入口
TrainerConfig 配置
TrainerConfig 中由当前 Trainer 主路径使用的配置区域如下:
| 配置字段 | 作用 | 相关文档 |
|---|---|---|
model / peft |
配置模型构建入口、预训练权重、数据类型、注意力实现和 PEFT | — |
training |
配置训练步数、batch、随机种子、梯度裁剪及日志和评估周期 | — |
accelerator / fsdp_config |
配置并行拓扑和 FSDP/HSDP 参数分片 | — |
mixed_precision |
配置 Trainer 混合精度开关 | — |
plan_overrides |
按模块匹配规则替换模型组件或覆盖并行计划 | — |
activation_checkpoint |
配置激活重计算 | Activation Checkpoint |
activation_swap / compile |
配置激活换入换出和模型编译 | — |
dataset / dataloader |
配置模型资产、样本转换、数据集读取、batch 组装和模型输入 | Dataset / DataLoader |
loss_fn / optimizer / lr_scheduler |
配置损失计算、参数分组、优化器和学习率策略 | — |
checkpoint |
配置模型、优化器和训练进度的保存与恢复 | Distributed Checkpoint |
debug / profiling / wandb |
配置数据与数值检查、性能采集和训练指标上报 | — |
字段解析、Target 和 CLI dotted override 规则见 YAML Trainer 配置结构;外部组件的接入方式见 二次开发指南。
模块结构
hyper_parallel/
├── models/ # 模型族适配层(adapter + recipes/train.yaml)、对外入口与 _transformers 模型构建
├── trainer/ # YAML 解析、TrainerConfig、组件构建、训练循环与生命周期
├── components/ # 优化器、损失函数、checkpoint、高性能 modules/functional 与量化
├── data/ # 数据集读取、样本转换与 dataloader
└── distributed/ # 分布式模型构建、并行计划应用与激活管理
模型族的替换工厂、CP/EP 规则和训练入口收敛在 models/<family>/ 下;当前交付
models/qwen3_moe/recipes/train.yaml、models/deepseek_v32/recipes/train.yaml 与
models/kimi_k26/recipes/train.yaml(Kimi-K2.6 多模态,4 超节点 / 512 卡主线配方,
家族说明见 models/kimi_k26/README.md)。