Harrysunshine/so101-smolvla-lora-sim-real-10task
028
SO-101 SmolVLA LoRA(仿真 + 真机混训,10 个任务)
在 lerobot/smolvla_base 上做 LoRA 微调,训练集是仿真与真机混成的一份: 仿真三个抓放任务 1498 集 + 公开的 SO-101 真机遥操 9 个任务 2200 集 = 3698 集 / 128 万帧 / 10 个不同任务。两侧共有的 Pick up a cube 与 Pick up a can 指令文本逐字相同, 落在同一个任务号下。
三个仿真场景各 50 局的验收
同一份数据、同一套评测下,全参微调(403M 可训练参数)是 94.0 / 90.0 / 96.0%。 这份适配器只训 9,851,728 个参数(占全模型 460M 的 2.1%)。
挂在哪些层上
上游 SmolVLA 的默认 target_modules 只挂动作专家的 q/v 投影,换机器人本体时学不动 —— 照默认跑到 8000 步,三个评测点全 0.0%。基座里可挂 q/v 的一共三族,这份权重三族全挂:
r=64;state_proj / action_in_proj / action_out_proj / action_time_mlp_in|out 这五个具身投影走 full_training_modules 全量训练,不做低秩近似。
相关
- 仿真数据集与产线源码:<https://huggingface.co/datasets/Harrysunshine/so101-sim-pickplace-v2>
- 全参微调的同款权重:<https://huggingface.co/Harrysunshine/so101-smolvla-sim-real-10task>
- 真机 9 个任务的数据:<https://modelscope.cn/datasets/zhuzhuangtian/so101-pick-place-tasks>
