File size: 572 Bytes
458b805 468fdcd 458b805 468fdcd 5a9a7eb 458b805 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
---
datasets:
- lerobot/aloha_sim_transfer_cube_scripted
base_model:
- lerobot/pi0
---
LeRobot version of Pi0 fine-tuned for 20k steps on the `lerobot/aloha_sim_transfer_cube_scripted_image` dataset.
Usage:
```python
policy = PI0Policy.from_pretrained("BrunoM42/pi0_aloha_transfer_cube")
action = policy.select_action(batch)
```
Run evaluation:
```python
python lerobot/scripts/eval.py --policy.path=BrunoM42/pi0_aloha_transfer_cube --env.type=aloha --env.task=AlohaTransferCube-v0 --env.episode_length=600 --eval.n_episodes=10 --eval.batch_size=10 --seed=42
``` |