6.3 SmolVLA 训练

6.3 SmolVLA 训练

⚠️ 训练需要 NVIDIA GPU，无法在树莓派或 Jetson Nano 上运行。

SmolVLA 在 ACT 基础上增加语言理解能力。

显存-batch_size 对应

显存	建议 batch_size
24 GB	64
8 GB	28 以内

安装 SmolVLA 并训练

linux: pip install -e ".[smolvla]"
export HF_ENDPOINT=https://hf-mirror.com   # 中国大陆用户

lerobot-train \
    --dataset.repo_id=${HF_USER}/so101_test \
    --policy.push_to_hub=false \
    --policy.type=smolvla \
    --policy.device=cuda \
    --output_dir=outputs/train/smolvla_test \
    --job_name=smolvla_test \
    --batch_size=64 \
    --steps=20000 \
    --wandb.enable=false
macos: pip install -e ".[smolvla]"
export HF_ENDPOINT=https://hf-mirror.com   # 中国大陆用户

lerobot-train \
    --dataset.repo_id=${HF_USER}/so101_test \
    --policy.push_to_hub=false \
    --policy.type=smolvla \
    --policy.device=cuda \
    --output_dir=outputs/train/smolvla_test \
    --job_name=smolvla_test \
    --batch_size=64 \
    --steps=20000 \
    --wandb.enable=false
windows: pip install -e ".[smolvla]"
$env:HF_ENDPOINT="https://hf-mirror.com"   # 中国大陆用户

lerobot-train `
    --dataset.repo_id=${HF_USER}/so101_test `
    --policy.push_to_hub=false `
    --policy.type=smolvla `
    --policy.device=cuda `
    --output_dir=outputs/train/smolvla_test `
    --job_name=smolvla_test `
    --batch_size=64 `
    --steps=20000 `
    --wandb.enable=false