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