Add README
Browse files
README.md
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
datasets:
|
| 3 |
+
- lerobot/aloha_sim_transfer_cube_scripted
|
| 4 |
+
base_model:
|
| 5 |
+
- lerobot/pi0
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
LeRobot version of Pi0 fine-tuned for 20k steps on the `lerobot/aloha_sim_transfer_cube_scripted_image` dataset.
|
| 9 |
+
|
| 10 |
+
Usage:
|
| 11 |
+
```
|
| 12 |
+
policy = PI0Policy.from_pretrained("BrunoM42/pi0_aloha_transfer_cube")
|
| 13 |
+
action = policy.select_action(batch)
|
| 14 |
+
```
|
| 15 |
+
|
| 16 |
+
Run evaluation:
|
| 17 |
+
```
|
| 18 |
+
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 --eval.seed=42
|
| 19 |
+
```
|