评估
来自Waveshare Wiki
|
| ||||||||||||||||||||
| ||||||||||||||||||||
LeRobot 教程目录
8. 评估
- 注意:以下操作均在容器内执行
- 注意:需要进入到 lerobot 目录下执行以下代码
8.1 推理模型准备
8.1.1 压缩
训练后,推理模型的生成位置,以 "/home/ws/lerobot/outputs/train/act_roarm_m3_test" 为例:
tar czvf act_roarm_m3_test.tar.gz outputs/train/act_roarm_m3_test/checkpoints/100000
接着,选中文件,下载到本地。
8.1.2 解压
选中文件,上传到推理环境的 "/home/ws/lerobot" 目录。
将打包后的数据集放置到进行推理的容器内,进入到根目录,解压到"/home/ws/lerobot":
cd /home/ws/lerobot && tar xzvf act_roarm_m3_test.tar.gz
进入 /home/ws/lerobot 目录,为模型创建链接:
ln -sfn 100000 outputs/train/act_roarm_m3_test/checkpoints/last
8.2 录制评估数据集
使用 lerobot/scripts/control_robot.py 中的函数,但使用策略检查点作为输入。例如,运行以下命令以录制 10 个评估集:
python lerobot/scripts/control_robot.py \
--robot.type=roarm_m3 \
--control.type=record \
--control.fps=30 \
--control.single_task="Grasp a block and put it in the bin." \
--control.repo_id=${HF_USER}/eval_act_roarm_m3_test \
--control.tags='["tutorial"]' \
--control.warmup_time_s=5 \
--control.episode_time_s=30 \
--control.reset_time_s=30 \
--control.num_episodes=10 \
--control.push_to_hub=true \
--control.policy.path=outputs/train/act_roarm_m3_test/checkpoints/last/pretrained_model
按照官方教程,了解有关使用 LeRobot 控制真实机器人的更深入教程。
