Update README.md

README.md

@@ -24,13 +24,35 @@ You can use this model by installing LeRobot from [this branch](https://github.c
 To train the model:
 
 ```bash
-python lerobot/scripts/train.py policy=dot_pusht_image env=pusht
+python lerobot/scripts/train.py \
+    --policy.type=dot \
+    --dataset.repo_id=lerobot/pusht \
+    --env.type=pusht \
+    --env.task=PushT-v0 \
+    --output_dir=outputs/train/pusht_images \
+    --batch_size=24  \
+    --log_freq=1000 \
+    --eval_freq=10000 \
+    --save_freq=50000 \
+    --offline.steps=1000000 \
+    --seed=100000 \
+    --wandb.enable=true \
+    --num_workers=24 \
+    --use_amp=true \
+    --device=cuda \
+    --policy.return_every_n=2
 ```
 
 To evaluate the model:
 
 ```bash
-python lerobot/scripts/eval.py -p IliaLarchenko/dot_pusht_images eval.n_episodes=1000 eval.batch_size=100 seed=1000000
+python lerobot/scripts/eval.py \
+    --policy.path=IliaLarchenko/dot_pusht_images \
+    --env.type=pusht \
+    --env.task=PushT-v0 \
+    --eval.n_episodes=1000 \
+    --eval.batch_size=100 \
+    --seed=1000000
 ```
 
 Model size: