Views
No views yet
| Property | Value |
|---|---|
| Architecture | ACT (Action Chunking with Transformers) |
| Parameters | 52M |
| Task | ALOHA TransferCube-v0 |
| Training Steps | 60,000 |
| Batch Size | 32 |
| Success Rate | ~42% |
pip install lerobot gym-aloha1lerobot-train \
2 --policy.type=act \
3 --dataset.repo_id=lerobot/aloha_sim_transfer_cube_human_image \
4 --env.type=aloha \
5 --env.task=AlohaTransferCube-v0 \
6 --batch_size=32 \
7 --steps=60000 \
8 --eval_freq=5000 \
9 --output_dir=./outputs/act_aloha_cube_best \
10 --wandb.enable=false \
11 --policy.push_to_hub=false1lerobot-eval \
2 --policy.path=LeTau/act_aloha_transfer_cube \
3 --env.type=aloha \
4 --env.task=AlohaTransferCube-v0 \
5 --eval.batch_size=1 \
6 --eval.n_episodes=201lerobot-train \
2 --resume=true \
3 --config_path=LeTau/act_aloha_transfer_cube/train_config.json \
4 --steps=100000| Evaluation | Episodes | Success Rate | Avg Sum Reward |
|---|---|---|---|
| Training | 50 | 42% | 116.26 |
| Independent | 20 | 35% | 95.95 |
Sum Rewards: [0.0, 241.0, 57.0, 201.0, 48.0, 0.0, 0.0, 220.0, 262.0, 0.0,
59.0, 211.0, 287.0, 187.0, 74.0, 2.0, 203.0, 18.0, 10.0, 0.0,
0.0, 263.0, 7.0, 57.0, 39.0, 214.0, 297.0, 24.0, 0.0, 274.0,
201.0, 2.0, 228.0, 228.0, 68.0, 290.0, 2.0, 222.0, 31.0, 219.0,
69.0, 22.0, 0.0, 76.0, 244.0, 227.0, 0.0, 26.0, 192.0, 211.0]
Successes: 21/50 episodes1@article{zhao2023learning,
2 title={Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware},
3 author={Zhao, Tony Z and Kumar, Vikash and Levine, Sergey and Finn, Chelsea},
4 journal={arXiv preprint arXiv:2304.13705},
5 year={2023}
6}