wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix revision

This commit is contained in:
Alexander Soare 2024-03-27 13:22:14 +00:00
parent 011f2d27fe
commit 6cd671040f
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -141,7 +141,6 @@ Or you can achieve the same result by executing our script from the command line
```bash ```bash
python lerobot/scripts/eval.py \ python lerobot/scripts/eval.py \
--hub-id lerobot/diffusion_policy_pusht_image \ --hub-id lerobot/diffusion_policy_pusht_image \
--revision v1.0 \
eval_episodes=10 \ eval_episodes=10 \
hydra.run.dir=outputs/eval/example_hub hydra.run.dir=outputs/eval/example_hub
``` ```

2
lerobot/scripts/eval.py
View File

@ -249,7 +249,7 @@ if __name__ == "__main__":
# TODO(alexander-soare): Save and load stats in trained model directory. # TODO(alexander-soare): Save and load stats in trained model directory.
stats_path = None stats_path = None
elif args.hub_id is not None: elif args.hub_id is not None:
folder = Path(snapshot_download(args.hub_id, revision="v1.0")) folder = Path(snapshot_download(args.hub_id, revision=args.revision))
cfg = init_hydra_config( cfg = init_hydra_config(
folder / "config.yaml", [f"policy.pretrained_model_path={folder / 'model.pt'}", *args.overrides] folder / "config.yaml", [f"policy.pretrained_model_path={folder / 'model.pt'}", *args.overrides]
) )