跳到主要内容

策略推理


1. 策略推理 (Inference)

使用训练好的策略模型进行推理并自动录制结果:

/isaac-sim/python.sh src/lerobot/scripts/lerobot_record.py \
--robot.type=walker_s2_sim \
--robot.headless=false \
--task=Part_Sorting \
--policy.path=your_checkpoint_dir \
--dataset.repo_id=your_org/eval_Part_Sorting \
--dataset.single_task="Part Sorting" \
--dataset.num_episodes=2 \
--dataset.push_to_hub=false \
--dataset.episode_time_s=100000000 \
--dataset.num_image_writer_processes=4 \
--dataset.root=${workspaceFolder}/datasets/Part_Sorting/your_policy_type \
--dataset.video=true \
--play_sounds=false
参数说明默认值 / 备注
--robot.type机器人类型walker_s2_sim
--robot.headless是否无头模式false
--task任务名称(Part_Sorting, Conveyor_Sorting, Foam_Inlaying, Packing_Box)Foam_Inlaying
--policy.path策略模型检查点路径(本地或 HuggingFace)必填
--dataset.repo_id数据集标识符,推理模式应为 xx/eval_xx 格式必填
--dataset.single_task任务描述必填(或由 task 自动设置)
--dataset.num_episodes推理回合数50
--dataset.root推理结果保存路径必填
--dataset.video是否录制视频true
--dataset.fps帧率30
--dataset.episode_time_s每回合时长(秒)60
--dataset.push_to_hub是否上传到 Hugging Facefalse
--play_sounds是否播放语音提示true

注意:

  1. 策略推理使用 lerobot_record.py 脚本,通过 --policy.path 参数指定策略模型路径。脚本会自动加载模型并在仿真环境中执行推理,同时录制结果数据。
  2. 推理模式下,--dataset.repo_id 参数应使用 xx/eval_xx 格式(如 your_org/eval_Part_Sorting),以区别于数据采集模式下的正式数据集命名。
  3. ${workspaceFolder} 应替换为实际的工作目录路径。