Transformers
Safetensors
Inference Endpoints
alexandersoare commited on
Commit
b9c3a19
1 Parent(s): b3e47e2

Update README.md

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