6.1 模型选择
6.1 模型选择
⚠️ 重要提示:模型训练需要 NVIDIA GPU,无法在树莓派或 Jetson Nano 上运行。
全部可用策略
| 策略 | 类型 | 说明 | 推荐场景 |
|---|---|---|---|
act |
模仿学习 | Action Chunking Transformers | 新手首选,训练稳定 |
diffusion |
扩散策略 | Diffusion Policy,去噪生成动作 | 复杂动作序列,高精度 |
vqbet |
行为 Transformer | VQ-BeT,离散化动作 token | 多模态行为 |
smolvla |
VLA | 轻量级视觉-语言-动作模型 | 语言指令引导的任务 |
pi0 |
VLA | π0,23 亿参数流匹配模型 | 多任务泛化 |
硬件要求速查
| 模型 | 最低显存 | 推荐显存 | 训练时长 (10 组) |
|---|---|---|---|
| ACT | 6 GB | 8 GB+ | 1-2 小时 |
| Diffusion | 8 GB | 12 GB+ | 2-4 小时 |
| SmolVLA | 8 GB | 12 GB+ | 2-4 小时 |
| Pi0 | 16 GB | 24 GB+ | 4-8 小时 |
新手推荐路径:ACT → SmolVLA → Pi0。