跳到主要内容

Spirit-v1.5 模型微调

Spirit-v1.5 是 Spirit AI 的视觉-语言-动作(Vision-Language-Action,VLA)策略。平台以模型标识 spirit-v1.5 注册,使用 spirit 训练框架,支持 robochallenge、LeRobot v2 与 v3 数据集。

适用角色与前提​

适用角色​

角色是否可创建训练任务说明
管理员是不受模块权限限制
项目经理需授权由管理员在模块权限中授予 「创建训练任务」
标注员需授权同上
审核员需授权同上
采集员需授权同上

仅查看训练任务与指标需「查看训练」。

前置条件​

项要求
数据集版本robochallenge、LeRobot v2、v3。RoboChallenge 布局可直接训练,LeRobot 数据在训练前转换为 RoboChallenge 布局
框架spirit
显存无固定下限;建议多卡训练。默认每卡 batch_size 为 1
依赖平台的模型训练服务已启用并配置至少一个训练位置;训练镜像内置 Spirit-v1.5 基座权重
状态字段observation.state 至少 8 维:前 7 维为末端位姿,第 8 维为夹爪宽度
视频字段需提供三路 RGB 视频,映射到 main_realsense_rgb、handeye_realsense_rgb、side_realsense_rgb
权限创建需「创建训练任务」,查看需「查看训练」

关节角状态的数据不能直接训练 Spirit-v1.5,需先完成状态空间适配。

创建训练任务与参数​

操作步骤​

  1. 进入「模型」→「训练」,点击「创建训练任务」。
  2. 选择训练位置,可选本机 GPU 或管理员已启用的云训练位置。
  3. 在「选择模型」中选中 spirit-v1.5。
  4. 在「训练数据集」中选择数据来源:导出记录、已上传 LeRobot 数据集、下载链接或 HuggingFace。
  5. 在「训练参数」中设置参数,含义与默认值见下表。使用 LeRobot 数据且相机字段无法自动匹配时,填写 camera_map。
  6. 填写备注(可选)。
  7. 点击「创建训练任务」。

训练参数​

参数说明默认值取值范围
batch_size每卡批大小1整数,≥ 1
steps最大训练步数10000整数,≥ 1
save_intervalcheckpoint 保存间隔2500整数,大于 0 时下发
log_freq日志间隔25整数,≥ 1
num_workersDataLoader 进程数2整数,≥ 0
prefetch_factorDataLoader 预取因子2整数,≥ 1
num_gpustorchrun 进程数可见 GPU 数,否则 1正整数
wandb_mode指标上报模式disableddisabled、offline、online
pretrained_path基座权重目录镜像内置基座权重含 model.safetensors 与 config.json 的目录
task_nameLeRobot 转换时的任务名move_objects_into_box字符串
task_promptLeRobot 转换时的语言提示数据集任务文本字符串
camera_mapLeRobot 相机字段到 Spirit 三相机的映射自动推断JSON

结果校验​

训练监控​

训练详情页按 Spirit 日志格式解析指标并绘制曲线:

指标含义判断标准
step已完成训练步数逐步递增至设定的 steps
total_steps设定的训练总步数与 steps 一致
loss训练损失总体下降
learning_rate当前学习率按调度策略变化

校验清单​

训练服务按容器退出码判定任务终态:退出码为 0 记为成功,非 0 记为失败。

校验项通过标准
任务状态训练详情页状态为「成功」,容器退出码为 0
产物目录输出目录下的 spirit_train/
检查点「模型输出」列出 checkpoint;last 指向的条目类型为「最终」,其余为「步骤」
指标曲线「训练指标」绘制 step、loss、learning_rate
日志「实时日志」可读取,无持续报错

Spirit 的训练与推理分离,部署时使用平台推理服务。「模型输出」中的 checkpoint 可下载、同步到对象存储;推理侧说明见模型推理。

异常处置​

现象可能原因处置责任方
转换时提示缺少三路相机LeRobot 数据缺少可识别的相机字段在 camera_map 中显式映射三路相机字段算法工程师
转换时提示 observation.state 维度不足状态少于 8 维,或为关节角而非末端位姿提供 7 维末端位姿加 1 维夹爪宽度的状态向量算法工程师
权重目录不可用指定目录缺少 model.safetensors 或 config.json使用镜像内置基座权重,或指定完整目录管理员
显存不足batch_size 或 num_workers 过大降低 batch_size 与 num_workers;改用多卡训练算法工程师
训练立即失败,提示找不到数据集输入目录既不是 RoboChallenge 布局,也无法转换检查 meta/info.json 与目录结构后重新选择数据集项目经理
训练位置不可用平台未启用该训练位置,或该位置 GPU 不可用更换训练位置;联系管理员检查训练服务管理员

相关页面​

页面用途
模型训练训练任务的创建、监控与配额
LeRobot 数据集与训练数据导出与训练总览
LeRobot v2 与 v3 格式差异两种数据格式的差异与迁移
数据导出生成 LeRobot 训练包