🚀 Thanos拾取力量宝石模型卡

这个策略模型使用 LeRobot 进行训练并推送到了模型中心。完整的文档请参考 LeRobot 文档。

🚀 快速开始

若想获取完整的操作指南，请查看训练指南。以下是训练和运行推理/评估的简要步骤：

📦 安装指南

从头开始训练

python lerobot/scripts/train.py \
  --dataset.repo_id=<用户或组织>/<数据集> \
  --policy.type=act \
  --output_dir=outputs/train/<期望的策略仓库 ID> \
  --job_name=lerobot_training \
  --policy.device=cuda \
  --policy.repo_id=<用户或组织>/<期望的策略仓库 ID> \
  --wandb.enable=true

将检查点写入 outputs/train/<期望的策略仓库 ID>/checkpoints/。

评估策略

python -m lerobot.record \
  --robot.type=so100_follower \
  --dataset.repo_id=<用户或组织>/eval_<数据集> \
  --policy.path=<用户或组织>/<期望的策略仓库 ID> \
  --episodes=10

在数据集仓库前加上 eval_ 前缀，并提供指向本地或模型中心检查点的 --policy.path。

📄 许可证

本项目采用 Apache-2.0 许可证。