训练与部署内置模型

更新时间:
复制 MD 格式

本文介绍如何使用平台内置的 VLA 模型进行训练,并将训练完成的模型部署为推理服务。这是最简路径——无需创建开发机,无需自定义模型,直接使用内置模型完成训练和部署。

模型库

平台内置以下主流具身智能模型:

模型

说明

GR00T

NVIDIA 通用机器人基础模型,支持多模态理解和任务执行。

PI 0.5

具身智能策略模型,专为机器人控制和决策设计。

在左侧导航栏中,展开模型管理,单击模型库可浏览所有可用模型。

训练内置模型

  1. 在左侧导航栏中,单击模型训练

  2. 单击开始训练

  3. 在弹出的对话框中,确认模型来源官方模型(默认选中)。

  4. 配置以下参数:

    参数

    是否必填

    说明

    数据集

    必填

    选择用于训练的数据集。需提前在数据集页面注册数据集。

    基础模型类型

    必填

    从下拉列表中选择模型类型(gr00tpi0)。

    选择模型

    必填

    选定模型版本。

    训练参数

    必填

    配置 batch size、最大训练步数、检查点保存间隔等训练参数。

  5. 单击确定,训练任务开始执行。

监控训练过程

训练任务提交后,可在模型训练列表中查看任务状态和进度:

说明

模型名称

训练产出模型的名称。

基础模型类型

GR00T 或 PI 0.5。

数据集名称

训练使用的数据集。

Ray Dashboard

单击链接可查看训练日志和资源使用情况。

状态

训练中、已完成或失败。

说明

训练指标(Loss 曲线、学习率等)通过训练任务列表的更多 > 查看指标入口查看,需在提交训练时填写 TensorBoard 日志路径。

部署推理服务

训练完成的模型可一键部署为云端推理服务。

  1. 在左侧导航栏中,单击模型部署

  2. 在模型列表中找到状态为空闲的已训练模型。

  3. 单击部署推理服务

部署过程自动执行,无需额外配置。平台基于 Ray Serve 实现模型推理服务部署,可通过对应的 Ray Dashboard 查看部署日志、GPU 资源占用和服务运行状态。

部署完成后:

  • 模型状态从"空闲"变为"已部署"。

  • 推理服务 URL 列显示推理服务的访问地址。

说明

无论部署的是 GR00T 还是 Pi0 模型,推理服务部署时均会占用 1 个 GPU 资源。

如需停止推理服务,单击停止推理服务,系统将停止当前推理服务并释放所占用的 GPU 资源。

页面顶部的统计卡片显示各状态的模型数量:已部署、部署中、空闲和其他。

端云协同推理

当推理服务处于已部署状态后,可通过平台将云端推理服务接入机器人运行流程,实现端云协同推理。对于已兼容 LeRobot 的机器人,可快速打通端云协同推理链路;对于其他控制协议或数据格式,可在 adb-robot-client SDK 基础上进行二次开发。

操作流程

  1. 进入机器人部署页面:在模型部署页面,单击已部署状态的模型,选择机器人部署

  2. 添加机器人设备:单击添加设备,选择已注册到平台的在线机器人。系统会自动检测设备在线状态。

  3. 初始化设备:单击连接,平台向机器人发送推理服务连接信息和训练数据集的第一帧数据(用于初始化机器人状态)。初始化成功后推理状态变为就绪

  4. 开始推理:单击开始推理,机器人持续向推理服务发送请求,开始端云协同推理。

  5. 停止推理:单击重置,停止推理流程并恢复初始状态。

说明

端云协同推理需要在线设备。设备接入方法请参见管理设备与机器人接入

相关文档