Update README.md
Browse files
README.md
CHANGED
@@ -18,7 +18,27 @@ See the [LeRobot library](https://github.com/huggingface/lerobot) (particularly
|
|
18 |
|
19 |
Trained with [LeRobot@d747195](https://github.com/huggingface/lerobot/tree/d747195c5733c4f68d4bfbe62632d6fc1b605712).
|
20 |
|
21 |
-
The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [aloha_sim_transfer_cube_human](https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_human/tree/v1.3) dataset
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
|
23 |
The training curves may be found at https://wandb.ai/alexander-soare/Alexander-LeRobot/runs/hjdard15.
|
24 |
|
|
|
18 |
|
19 |
Trained with [LeRobot@d747195](https://github.com/huggingface/lerobot/tree/d747195c5733c4f68d4bfbe62632d6fc1b605712).
|
20 |
|
21 |
+
The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [aloha_sim_transfer_cube_human](https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_human/tree/v1.3) dataset, using this command:
|
22 |
+
|
23 |
+
```bash
|
24 |
+
python lerobot/lerobot/scripts/train.py \
|
25 |
+
hydra.job.name=act_aloha_sim_transfer_cube_human \
|
26 |
+
hydra.run.dir=outputs/train/act_aloha_sim_transfer_cube_human \
|
27 |
+
policy=act \
|
28 |
+
policy.use_vae=true \
|
29 |
+
env=aloha \
|
30 |
+
env.task=AlohaTransferCube-v0 \
|
31 |
+
dataset_repo_id=lerobot/aloha_sim_transfer_cube_human \
|
32 |
+
training.eval_freq=10000 \
|
33 |
+
training.log_freq=250 \
|
34 |
+
training.offline_steps=100000 \
|
35 |
+
training.save_model=true \
|
36 |
+
training.save_freq=25000 \
|
37 |
+
eval.n_episodes=50 \
|
38 |
+
eval.batch_size=50 \
|
39 |
+
wandb.enable=true \
|
40 |
+
device=cuda
|
41 |
+
```
|
42 |
|
43 |
The training curves may be found at https://wandb.ai/alexander-soare/Alexander-LeRobot/runs/hjdard15.
|
44 |
|