Robotics
Transformers
Safetensors
Inference Endpoints
alexandersoare commited on
Commit
c887ff6
1 Parent(s): 0ffb2d2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -17,7 +17,25 @@ See the [LeRobot library](https://github.com/huggingface/lerobot) (particularly
17
 
18
  ## Training Details
19
 
20
- The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [pusht](https://huggingface.co/datasets/lerobot/pusht/tree/v1.3) dataset.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  The training curves may be found at https://wandb.ai/alexander-soare/Alexander-LeRobot/runs/508luayd.
23
 
 
17
 
18
  ## Training Details
19
 
20
+ The model was trained using [LeRobot's training script](https://github.com/huggingface/lerobot/blob/d747195c5733c4f68d4bfbe62632d6fc1b605712/lerobot/scripts/train.py) and with the [pusht](https://huggingface.co/datasets/lerobot/pusht/tree/v1.3) dataset, using this command:
21
+
22
+ ```bash
23
+ python lerobot/scripts/train.py \
24
+ hydra.run.dir=outputs/train/diffusion_pusht \
25
+ hydra.job.name=diffusion_pusht \
26
+ policy=diffusion training.save_model=true \
27
+ env=pusht \
28
+ env.task=PushT-v0 \
29
+ dataset_repo_id=lerobot/pusht \
30
+ training.offline_steps=200000 \
31
+ training.save_freq=20000 \
32
+ training.eval_freq=10000 \
33
+ eval.n_episodes=50 \
34
+ wandb.enable=true \
35
+ wandb.disable_artifact=true \
36
+ device=cuda
37
+ ```
38
+
39
 
40
  The training curves may be found at https://wandb.ai/alexander-soare/Alexander-LeRobot/runs/508luayd.
41