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)。