8.3 双臂模型训练
8.3 双臂模型训练
双臂模型训练命令与单臂基本相同,区别仅在使用双臂采集的数据集:
lerobot-train \
--dataset.repo_id=test/bimanual-so101-test \
--policy.type=act \
--output_dir=outputs/train/act_bimanual_test \
--job_name=act_bimanual_test \
--policy.device=cuda \
--policy.push_to_hub=false \
--wandb.enable=false
训练配置(模型选择、超参数)与单臂通用。建议先短训练验证流程,确认无误后再拉长训练。