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 |
|