跳到主要内容

概述

本文档汇总 pi0 / pi05(微调 ckpt 030000)从 LeRobot 训练、到 S600 量化编译、再到板端部署运行的完整链路,
以及各阶段已遇到的问题与解决方法。

pi0 与 pi05 差异说明​

对比项pi05pi0
version0.50
state 注入无 token;
走 AdaRMSNorm(adarms) 调制
state_proj(32→1024) + 前缀 state token
time 注入time_mlp → adarms 调制action_time_mlp 与 action 拼接进输入
prompt 长度20048
action suffix[action_horizon](50)[state_token] + [action_horizon](51)
引擎 time_horizon_01
enable_time_mod_luttruefalse
运行时 state 注入inject_state=true(prompt 注入)inject_state=false(走 state 张量)