update readme

This commit is contained in:
Alexander Soare 2024-03-22 10:34:22 +00:00
parent 8720c568d0
commit 3f0f95f4c0
1 changed files with 5 additions and 10 deletions

View File

@ -48,7 +48,6 @@ wandb login
## Usage ## Usage
### Train ### Train
``` ```
@ -65,14 +64,9 @@ hydra.run.dir=tmp/$(date +"%Y_%m_%d") \
env=pusht env=pusht
``` ```
### Visualize online buffer / Eval ### Eval
```
python lerobot/scripts/eval.py \
hydra.run.dir=tmp/$(date +"%Y_%m_%d") \
env=pusht
```
Run `python lerobot/scripts/eval.py --help` for instructions.
## TODO ## TODO
@ -106,8 +100,9 @@ with profile(
```bash ```bash
python lerobot/scripts/eval.py \ python lerobot/scripts/eval.py \
pretrained_model_path=/home/rcadene/code/fowm/logs/xarm_lift/all/default/2/models/final.pt \ --config /home/rcadene/code/fowm/logs/xarm_lift/all/default/2/.hydra/config.yaml \
eval_episodes=7 pretrained_model_path=/home/rcadene/code/fowm/logs/xarm_lift/all/default/2/models/final.pt \
eval_episodes=7
``` ```
## Contribute ## Contribute