Model Card for AI Worker Demonstration

This repository provides a pretrained action chunking transformer policy trained for ROBOTIS AI Worker demonstration.

Training Details

The model was trained using LeRobot's training script and aiworker_whole_gripper.AdapterCableToBasket dataset, using this command:

python lerobot/scripts/train.py \
  --dataset.repo_id=ROBOTIS/aiworker_whole_gripper.AdapterCableToBasket \
  --policy.type=act \
  --output_dir=outputs/train/aiworker_whole_gripper.AdapterCableToBasket \
  --policy.device=cuda \
  --wandb.enable=true