alexandersoare
commited on
Commit
•
c887ff6
1
Parent(s):
0ffb2d2
Update README.md
Browse files
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 |
|