BrunoM42 commited on
Commit
458b805
·
verified ·
1 Parent(s): 49bafe3

Add README

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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
+ ```